search

JuliaStats / MLBase.jl

A set of functions to support the development of machine learning algorithms
MIT License
185 stars 63 forks source link

Add StratifiedKfold and StratifiedRandomSub #6

Closed lendle closed 10 years ago

lendle commented 10 years ago

I also added unique_inverse in utils.jl which returns a vector of unique elements of an array and a vector of vectors of their indexes, but did not export it.

Note that the way StratifiedKfold works is not consistent with Kfold in master right now, but is consistent with Kfold in the pull request in #5. That is, if #5 is merged, Kfold(n, k) and StratifiedKfold(fill(1, n), k) will do the same thing.

lendle commented 10 years ago

Forgot to add, this would close #3.