wassname/multifit
1
0
Fork 0
mirror of https://github.com/wassname/multifit.git synced 2026-09-11 12:20:41 +08:00
Code Issues Packages Projects Releases Wiki Activity

multiclass (but single classes don't work)

Browse Source
This commit is contained in:
wassname
2019-11-26 12:19:32 +08:00
parent 7557805043
commit 46ec76ec78
4 changed files with 77 additions and 22 deletions
Download Patch File Download Diff File Expand all files Collapse all files
fastai_contrib/data_block.py
+6 -5
View File
@@ -16,11 +16,12 @@ class BinaryCategoryList(CategoryListBase):
def __init__(self, items:Iterator, classes:Collection=None, label_delim:str=None, **kwargs):
super().__init__(items, classes=classes, **kwargs)
mean = self.items.mean()
if mean and mean!=0:
weight = torch.tensor([1 / self.items.mean()]).cuda()
else:
weight = None
raise Exception('debug')
# if mean and mean != 0:
# weight = torch.tensor([1 / mean]).cuda()
# print(f'Weighting BCEWithLogitsFlat by {weight.item()}')
# else:
weight = None
# raise Exception('debug')
self.loss_func = BCEWithLogitsFlat(weight=weight)
def reconstruct(self, t):