From 38e131f802744de17a83c7004b450ff8c4c4fe08 Mon Sep 17 00:00:00 2001 From: Piotr Czapla Date: Sun, 19 May 2019 00:17:03 +0200 Subject: [PATCH] Make the ulmfit compatible with old fastai (ulmfit_multilingual) --- ulmfit/train_clas.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/ulmfit/train_clas.py b/ulmfit/train_clas.py index 1c926f8..3892dc9 100644 --- a/ulmfit/train_clas.py +++ b/ulmfit/train_clas.py @@ -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