Add support for stratified CV with one-hot encoding