Есть ли ограничение на тип данных при использовании train_test_split. У меня есть набор данных, в котором все особенности «X» являются целыми числами, а целевой «Y» является категоричным ... в этом случае train_test_split выдает ошибку take_nd (). Но когда я изменяю 'Y' на int, он работает нормально.