I was running cross-validation in parallel on a cluster using cross_validate_sustain_model() with argument select_fold set to the CV fold desired for each compute job.
I noticed that all 10 folds were returning results for only fold0.
The culprit is line 276, where the loop is through range(Nfolds) (where Nfolds=len(select_fold)) rather than explicitly through the select_fold array itself.
Will send a PR to fix shortly, but wanted to raise this in case others have the same problem
noxtoby
commented
2 years ago
I was running cross-validation in parallel on a cluster using
cross_validate_sustain_model()with argumentselect_foldset to the CV fold desired for each compute job.I noticed that all 10 folds were returning results for only
fold0.The culprit is line 276, where the loop is through
range(Nfolds)(whereNfolds=len(select_fold)) rather than explicitly through theselect_foldarray itself.Will send a PR to fix shortly, but wanted to raise this in case others have the same problem