From 8facc21cfa2beed7fb895f1b8eb240b7ea3d994b Mon Sep 17 00:00:00 2001
From: Piotr Czapla <github@piotrczapla.com>
Date: Sun, 14 Apr 2019 19:13:43 +0200
Subject: [PATCH] Add F1 metrics to evaluate

---
 calc_100.sh          | 4 ----
 ulmfit/train_clas.py | 3 +++
 2 files changed, 3 insertions(+), 4 deletions(-)
 delete mode 100755 calc_100.sh

diff --git a/calc_100.sh b/calc_100.sh
deleted file mode 100755
index 158de54..0000000
--- a/calc_100.sh
+++ /dev/null
@@ -1,4 +0,0 @@
-#!/usr/bin/env bash
-
-LANGS
-for
\ No newline at end of file
diff --git a/ulmfit/train_clas.py b/ulmfit/train_clas.py
index 417f791..b230387 100644
--- a/ulmfit/train_clas.py
+++ b/ulmfit/train_clas.py
@@ -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)