Update stats.py

powerfulbean · Jun 17, 2024 · 65ecbcf · 65ecbcf
1 parent d60e8cd
commit 65ecbcf
Showing 1 changed file with 15 additions and 7 deletions.
diff --git a/mtrf/stats.py b/mtrf/stats.py
@@ -366,13 +366,21 @@ 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],
-            fs,
-            tmin,
-            tmax,
-            regularization_split_i,
+        # 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, 
+            regularization_split_i
         )
         _, t_metric_test = model.predict(
             [stimulus[i] for i in idx_test],