Make the ulmfit compatible with old fastai (ulmfit_multilingual)

2026-09-10 12:12:50 +08:00 · 2019-05-19 00:17:03 +02:00
parent 6f4db9de1a
commit 38e131f802
1 changed files with 1 additions and 1 deletions
@@ -99,7 +99,7 @@ class CLSHyperParams(LMHyperParams):
        f1_score = FBeta(beta=1.0)
        precision = Precision()
        recall = Recall()
-        kappa_lin = KappaScore('linear')
+        kappa_lin = KappaScore()
        matthews_correff = MatthewsCorreff()
        metrics = [f1_score, precision, recall, kappa_lin, matthews_correff]
        # TODO: fix this in fast.ai