Merge pull request #33 from powerfulbean/fix-nested_crossval

fix bug of decoder training in nested_crossval
powerfulbean · Jun 26, 2024 · 2bc5c49 · 2bc5c49
2 parents 53838c8 + 6a75d47
commit 2bc5c49
Showing 1 changed file with 11 additions and 3 deletions.
diff --git a/mtrf/stats.py b/mtrf/stats.py
@@ -266,9 +266,17 @@ def nested_crossval(
             regularization_split_i = list(regularization)[np.argmax(metric)]
         else:
             regularization_split_i = regularization
-        model.train(
-            [stimulus[i] for i in idx_train_val],
-            [response[i] for i in idx_train_val],
+        # model.train(
+        #     [stimulus[i] for i in idx_train_val],
+        #     [response[i] for i in idx_train_val],
+        #     fs,
+        #     tmin,
+        #     tmax,
+        #     regularization_split_i,
+        # )
+        model._train(
+            [x[i] for i in idx_train_val],
+            [y[i] for i in idx_train_val],
             fs,
             tmin,
             tmax,