Wait, what. Why are the labels getting distributed. It should retain parts of all labels for train and for validation right.
