Add F1 metrics to evaluate

This commit is contained in:
Piotr Czapla
2019-04-14 19:13:43 +02:00
parent a948d7df91
commit 8facc21cfa
2 changed files with 3 additions and 4 deletions
-4
View File
@@ -1,4 +0,0 @@
#!/usr/bin/env bash
LANGS
for
+3
View File
@@ -110,6 +110,9 @@ class CLSHyperParams(LMHyperParams):
learn = self.create_cls_learner(data_tst, drop_mult=0.3, label_smoothing_eps=label_smoothing_eps)
learn.unfreeze()
learn.load(save_name)
f1 = FBeta(beta=1, average='binary')
f1.on_train_begin()
learn.metrics += [f1]
val_res=[-1, -1]
if data_cls:
val_res = learn.validate(data_cls.valid_dl)