fix evaluate

ppdebreuck · Nov 13, 2021 · 82e83f9 · 82e83f9
1 parent 938a3de
commit 82e83f9
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/modnet/models/vanilla.py b/modnet/models/vanilla.py
@@ -654,7 +654,7 @@ def evaluate(self, test_data: MODData) -> pd.DataFrame:
             x = self._scaler.transform(x)
             x = np.nan_to_num(x, nan=-1)
 
-        y_pred = self.model.predict(x)
+        y_pred = np.array(self.model.predict(x))
         if len(y_pred.shape) == 2:
             y_pred = np.array([y_pred])
         score = []