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

Add workaround for metrics init

Browse Source
This commit is contained in:
Marcin
2019-05-12 21:43:54 +02:00
parent 6aff0a9e22
commit 50b73510de
1 changed files with 4 additions and 3 deletions
Download Patch File Download Diff File Expand all files Collapse all files
ulmfit/train_clas.py
+4 -3
View File
@@ -79,14 +79,15 @@ class CLSHyperParams(LMHyperParams):
if num_cls_epochs > 5:
learn.fit_one_cycle(num_cls_epochs-4, slice(1e-2 / (2.6 ** 4), 1e-2), moms=(0.8, 0.7), wd=1e-7)
def get_metrics(self):
def get_metrics(self, init=False):
f1_score = FBeta(beta=1.0)
precision = Precision()
recall = Recall()
metrics = [f1_score, precision, recall]
# TODO: fix this in fast.ai
for metric in metrics: metric.on_train_begin()
if init:
for metric in metrics: metric.on_train_begin()
metrics.append(accuracy)
return metrics
@@ -151,7 +152,7 @@ class CLSHyperParams(LMHyperParams):
if data_tst is None:
data_clas , _, data_tst = self.load_cls_data(bs)
if learn is None:
learn = self.create_cls_learner(data_tst, drop_mult=0.3, metrics=self.get_metrics())
learn = self.create_cls_learner(data_tst, drop_mult=0.3, metrics=self.get_metrics(True))
learn.unfreeze()
learn.load(save_name)
probs, targets = learn.get_preds(ordered=True)