Balanced minibatches using crossvalind()