From 6927566ddea7bcc5b88640f585aa7b44558615ba Mon Sep 17 00:00:00 2001
From: Nabeel <khnabeelasif@qlik.com>
Date: Fri, 8 Nov 2019 09:24:05 +1100
Subject: [PATCH] Fix for issue #65

Fix for sklearn when using a hold-out testing strategy. This bug was introduced with version 6.0.
---
 core/_sklearn.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/core/_sklearn.py b/core/_sklearn.py
index e736b56..4d68e73 100644
--- 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()