Julia Version 1.7.1
Commit ac5cc99908 (2021-12-22 19:35 UTC)
Platform Info:
OS: Linux (x86_64-pc-linux-gnu)
CPU: Intel(R) Core(TM) i5-9400 CPU @ 2.90GHz
WORD_SIZE: 64
LIBM: libopenlibm
LLVM: libLLVM-12.0.1 (ORCJIT, skylake)
using ScikitLearn.CrossValidation: train_test_split, StratifiedKFold, KFold
kf = KFold(10, n_folds=5, shuffle=true) # all is well
skf = StratifiedKFold([rand([-1, 1]) for i = 1:20], n_folds=5) # all is well
skf = StratifiedKFold([rand([-1, 1]) for i = 1:20], n_folds=5, shuffle=true) # error below
ScikitLearn v0.6.4
Julia Version 1.7.1 Commit ac5cc99908 (2021-12-22 19:35 UTC) Platform Info: OS: Linux (x86_64-pc-linux-gnu) CPU: Intel(R) Core(TM) i5-9400 CPU @ 2.90GHz WORD_SIZE: 64 LIBM: libopenlibm LLVM: libLLVM-12.0.1 (ORCJIT, skylake)