Potential Bug: train/test shape mismatch for multi-class classification

[gkirgizov]

В ходе investigation #947 я обнаружил такую потенциальную проблему. Если в тестовую и тренировочную выборки случайно попадают разные количества target классов (что становится более вероятным при небольших выборках, большом количестве классов и их несбалансированности), то train_data.num_classes != test_data.num_classes, что ведет к ошибке несоответствия размерностей в модели при валидации.

Нужно как-то иначе считать data.num_classes, так, чтобы учитывать весь набор данных.