Test correction

w4k2 · Dec 18, 2019 · e58ee07 · e58ee07
1 parent 17c7652
commit e58ee07
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/strlearn/tests/test_evaluators.py b/strlearn/tests/test_evaluators.py
@@ -3,7 +3,7 @@
 import sys
 import strlearn as sl
 from sklearn.metrics import accuracy_score, roc_auc_score
-from ..utils import bac, f_score, geometric_mean_score
+from ..utils import bac, f1_score, geometric_mean_score_1
 
 sys.path.insert(0, "../..")
 
@@ -24,7 +24,7 @@ def test_TTT_single_clf():
 def test_TTT_custom_metrics():
     stream = get_stream()
     clf = sl.classifiers.AccumulatedSamplesClassifier()
-    metrics = [accuracy_score, roc_auc_score, geometric_mean_score, bac, f_score]
+    metrics = [accuracy_score, roc_auc_score, geometric_mean_score_1, bac, f1_score]
     evaluator = sl.evaluators.TestThenTrain(metrics=metrics)
     evaluator.process(stream, clf)
 
@@ -45,7 +45,7 @@ def test_TTT_multiple_clfs():
         sl.classifiers.AccumulatedSamplesClassifier(),
         sl.classifiers.AccumulatedSamplesClassifier(),
     ]
-    metrics = [accuracy_score, roc_auc_score, geometric_mean_score, bac, f_score]
+    metrics = [accuracy_score, roc_auc_score, geometric_mean_score_1, bac, f1_score]
     evaluator = sl.evaluators.TestThenTrain(metrics=metrics)
     evaluator.process(stream, clfs)
 
@@ -58,7 +58,7 @@ def test_P_multiple_clfs():
         sl.classifiers.AccumulatedSamplesClassifier(),
         sl.classifiers.AccumulatedSamplesClassifier(),
     ]
-    metrics = [accuracy_score, roc_auc_score, geometric_mean_score, bac, f_score]
+    metrics = [accuracy_score, roc_auc_score, geometric_mean_score_1, bac, f1_score]
     evaluator = sl.evaluators.Prequential(metrics=metrics)
     evaluator.process(stream, clfs)
 

diff --git a/strlearn/tests/test_metrics.py b/strlearn/tests/test_metrics.py
@@ -30,7 +30,7 @@ def test_precision_recall():
 
             precision = sl.utils.metrics.precision(y_test, y_pred)
             recall = sl.utils.metrics.recall(y_test, y_pred)
-            gmean = sl.utils.metrics.geometric_mean_score(y_test, y_pred)
+            gmean = sl.utils.metrics.geometric_mean_score_1(y_test, y_pred)
 
         clf.partial_fit(chunk[0], chunk[1])