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