Merge pull request #17 from dataiku/bugfix/binary-classification-v11

fix binary classification prediction from v11

Author: david-behar

CHANGELOG.md

@@ -1,5 +1,9 @@
 # Changelog
 
+## [Version 1.0.2] - Bugfix Release - 2022-07
+
+* Fixed binary classification prediction output format to fit v11 stricter format
+
 ## [Version 1.0.1] - Bugfix Release - 2022-04
 
 * Fixed package requirement dependency issue with python 3.7 for the custom view

plugin.json

@@ -1,6 +1,6 @@
 {
     "id": "generalized-linear-models",
-    "version": "1.0.1",
+    "version": "1.0.2",
     "meta": {
         "label": "Generalized Linear Models",
         "description": "Train and deploy Generalized Linear Models",

python-lib/generalized_linear_models/dku_glm.py

@@ -291,9 +291,8 @@ def predict(self, X):
         Returns the binary target
         """
         y_pred = self.predict_target(X)
-        y_pred_final = y_pred.reshape((len(y_pred), -1))
 
-        return y_pred_final > 0.5
+        return y_pred > 0.5
 
     def predict_proba(self, X):
         """