remove one_hot, first try

I think all cases of the dataset must have the same number of classes, as `y` is a matrix.
I have to check what we should do if some indices are repeated.