Merge pull request #37 from ramp-kits/scoring

add dim to X in predict
ramp-kits · Dec 15, 2023 · de62e0d · de62e0d
2 parents f2f64e9 + b5cb5a1
commit de62e0d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/stroke/scoring.py b/stroke/scoring.py
@@ -52,8 +52,8 @@ def score_function(self, Y_true: np.array, Y_pred: np.array):
         fscore = 0
         for y_true_i, prediction_object in zip(y_true, Y_pred.y_pred):
             dat = estimator.predict(
-                BIDSLoader.load_image_tuple(prediction_object.pred)
-            )
+                BIDSLoader.load_image_tuple(prediction_object.pred)[None]
+            )[0]
 
             # Using proxy of y_true.shape != y_pred.shape to indicate that data needs to be unpacked
             if y_true_i.shape != dat.shape: