Неверные классы, выведенные из уникальных значений `y`. Ожидается: [0 1 2 3 4 5], получил [1 2 3 4 5 6] ⇐ Python

[Anonymous]

Я обучил набор данных, используя классификатор XGB, но я получил эту ошибку в Local. Это работало на Colab, а также у моих друзей нет проблем с тем же кодом.
Я не знаю, что означает эта ошибка ... < /p>
Invalid classes inferred from unique values of y. Expected: [0 1 2 3 4 5], got [1 2 3 4 5 6]
Это мой код, но я думаю, это не причина.
start_time = time.time()
xgb = XGBClassifier(n_estimators = 400, learning_rate = 0.1, max_depth = 3)
xgb.fit(X_train.values, y_train)
print('Fit time : ', time.time() - start_time)


Подробнее здесь: https://stackoverflow.com/questions/719 ... -3-4-5-got