mirror of
https://github.com/wassname/multifit.git
synced 2026-10-29 13:20:17 +08:00
Make the ulmfit compatible with old fastai (ulmfit_multilingual)
This commit is contained in:
1 parent
6f4db9de1a
commit
38e131f802
1 file changed
+1
-1
@@ -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
|
||||
|
||||
Reference in new issue
Block a user