Skip to content

Commit

Permalink
kfold startification on first column
Browse files Browse the repository at this point in the history
ppdebreuck authored and ml-evs committed May 31, 2023
1 parent 10962d2 commit b3d5289
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion modnet/matbench/benchmark.py
Original file line number Diff line number Diff line change
@@ -35,7 +35,7 @@ def _mapArrayToInt(a):

ycv = data.df_targets.iloc[:, 0].map(_mapArrayToInt)
else:
ycv = data.df_targets
ycv = data.df_targets.values[:, 0]

kf = KFold(n_splits=n_splits, shuffle=True, random_state=MATBENCH_SEED)
kf_splits = kf.split(data.df_featurized, y=ycv)

0 comments on commit b3d5289

Please sign in to comment.