fixed predict_classes in predict

autonomio · Jan 28, 2022 · 2460100 · 2460100
1 parent 010a319
commit 2460100
Showing 1 changed file with 7 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