fix random state of groupshufflesplit

OxWearables · Mar 9, 2024 · 5b419d5 · 5b419d5
1 parent 9bca56d
commit 5b419d5
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/actinet/models.py b/src/actinet/models.py
@@ -76,7 +76,7 @@ def fit(
         t_splits = []
 
         if n_splits < 3:
-            splitter = GroupShuffleSplit(n_splits=n_splits)
+            splitter = GroupShuffleSplit(n_splits=n_splits, random_state=42)
             split_iterator = splitter.split(X, Y, groups)
         else:
             splitter = StratifiedGroupKFold(n_splits)