MLJBase.jl
MLJBase.jl copied to clipboard
`partition` actually *does* allow `shuffle=true` if `stratify != nothing`?
The docs of partition say that
stratify=nothing: if a vector is specified, the partition will match the stratification of the given vector. In that case,shufflecannot be false`.
However,
-
shuffleis internally set totrueand a shuffle performed if an RNG is supplied, even ifshuffle=false(i.e. the docs should probably also say don't supply an RNG either?) - I'm fairly confident, that the
stratifyvector is actually correctly indexed into (here) even ifshuffle=true. (Ideally, someone with more code familiarity would quickly look over this, too, though—me being fairly confident may not be enough for the standards of this repo.)
If I'm mistaken, then sorry for the noise.
If not, then it may make sense to update the docs (in either case)?
Thanks @dpaetzel for your post. I've not had a chance to look into this yet but thank you in advance for your patience.
@tlienart If you still remember, are you able to address this point?