reimplement stratified-split algorithm

[kjappelbaum]

a better algorithm is probably to do the split for every class, throw an error if it does not work, and then combine