florent/minigrida
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Quick fix on cvgenerator

Browse Source
This commit is contained in:
Florent Guiotte 2020-07-08 17:46:22 +02:00
parent a2bca010f5
commit 2af2656837
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
minigrida/cvgenerators/generic.py
View File

@ -22,4 +22,4 @@ class GroupStratifiedShuffleSplit(GroupKFold):
for train, test in splits: for train, test in splits:
sss_train, sss_test = next(StratifiedShuffleSplit(1, train_size=self.train_size, random_state=self.random_state).split(X[train], y[train], groups[train])) sss_train, sss_test = next(StratifiedShuffleSplit(1, train_size=self.train_size, random_state=self.random_state).split(X[train], y[train], groups[train]))
yield sss_train, test yield train[sss_train], test