fix summary for classifiers

Techtonique · Jul 14, 2024 · f6ebcd1 · f6ebcd1
1 parent 8e7eac7
commit f6ebcd1
Showing 1 changed file with 3 additions and 1 deletion.
diff --git a/learningmachine/base.py b/learningmachine/base.py
@@ -213,7 +213,7 @@ def score(self, X, y, scoring=None, **kwargs):
             return scoring_options[scoring](y, preds, **kwargs)
 
     def summary(self, X,                 
-                class_index,                                   
+                class_index = None,                                   
                 y = None,                                                 
                 cl = None,
                 show_progress = True):
@@ -222,6 +222,8 @@ def summary(self, X,
 
             if self.type == "classification":
 
+                assert class_index is not None, "For classifiers, 'class_index' must be provided"
+
                 assert y is not None, "the response 'y' must be provided for classifiers"     
 
                 return self.obj["summary"](X = r.matrix(FloatVector(X.ravel()),