Fix for issue #65

Fix for sklearn when using a hold-out testing strategy. This bug was introduced with version 6.0.
nabeel-oz · Nov 7, 2019 · 6927566 · 6927566
1 parent 4239382
commit 6927566
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/core/_sklearn.py b/core/_sklearn.py
@@ -719,7 +719,7 @@ def fit(self):
             # Select the dataset for calculating importances
             if self.model.validation == "hold-out":
                 X = self.X_test
-                y = self.y_test.values.ravel()
+                y = self.y_test # Already a numpy array after calculate_metrics
             else:
                 X = self.X_train
                 y = self.y_train.values.ravel()