name in the first call

Author: mgraffg

CompStats/interface.py

@@ -90,6 +90,7 @@ class Perf(object):
 
     """
     def __init__(self, y_true, *y_pred,
+                 name:str=None,
                  score_func=balanced_accuracy_score,
                  error_func=None,
                  num_samples: int=500,
@@ -100,8 +101,13 @@ def __init__(self, y_true, *y_pred,
         self.score_func = score_func
         self.error_func = error_func
         algs = {}
-        for k, v in enumerate(y_pred):
-            algs[f'alg-{k+1}'] = np.asanyarray(v)
+        if name is not None:
+            if isinstance(name, str):
+                name = [name]
+        else:
+            name = [f'alg-{k+1}' for k, _ in enumerate(y_pred)]
+        for key, v in zip(name, y_pred):
+            algs[key] = np.asanyarray(v)
         algs.update(**kwargs)
         self.predictions = algs
         self.y_true = y_true
@@ -519,7 +525,7 @@ def y_true(self, value):
                 algs[c] = value[c].to_numpy()
             self.predictions.update(algs)
             return
-        self._y_true = value
+        self._y_true = np.asanyarray(value)
 
     @property
     def score_func(self):

CompStats/tests/test_interface.py

@@ -23,6 +23,13 @@
 from CompStats.tests.test_performance import DATA
 
 
+def test_Perf_name():
+    """Test Perf name keyword"""
+    from CompStats.metrics import f1_score
+    score = f1_score([1, 0, 1], [1, 0, 0], name='algo')
+    assert 'algo' in score.predictions
+
+
 def test_Perf_plot_col_wrap():
     """Test plot when 2 classes"""
     from CompStats.metrics import f1_score