Разделение набора данных для классификации - PullRequest
0 голосов
/ 28 февраля 2020

Я пытаюсь обучить и протестировать модель классификации, однако я не понимаю, почему я получаю эту ошибку:

ValueError: The test_size = 9 should be greater or equal to the number of classes = 11

Что означает эта ошибка?

Мой код Для разделения наборов данных используются следующие:

# Splitting the Dataset:
X_train, _, y_train,_ = train_test_split(filtered_features_FM, y_Cat, stratify=y_Cat)
_, X_test, _, y_test = train_test_split(filtered_features_FM_T, y_T_Cat, stratify=y_T_Cat)

Ошибка возникает на тестовой линии.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...