validation set should contain sample for each class

[elahehraisi]

The split of labeled data to train and val should be managed so that validation set contains samples for every class (ideally the same number of samples per class).