Merge pull request #564 from autonomio/fix_predict

Fix predict
autonomio · Jan 28, 2022 · 744b654 · 744b654
2 parents 010a319 + dace33a
commit 744b654
Show file tree

Hide file tree

Showing 2 changed files with 11 additions and 1 deletion.
diff --git a/talos/commands/predict.py b/talos/commands/predict.py
@@ -44,11 +44,17 @@ def predict_classes(self, x, metric, asc, model_id=None):
 
         '''
 
+        import numpy as np
+
         if model_id is None:
             from ..utils.best_model import best_model
             model_id = best_model(self.scan_object, metric, asc)
 
         from ..utils.best_model import activate_model
         model = activate_model(self.scan_object, model_id)
 
-        return model.predict_classes(x)
+        # make (class) predictiosn with the model
+        preds = model.predict(x)   
+        preds_classes = np.argmax(preds, axis=1)
+
+        return preds_classes
diff --git a/tests/commands/test_latest.py b/tests/commands/test_latest.py
@@ -1,5 +1,9 @@
 def test_latest():
 
+    import warnings
+
+    warnings.simplefilter('ignore')
+
     print('\n >>> start Latest Features... \n')
 
     import talos