CatBoostError: catboost/private/libs/target/target_converter.cpp:228: Unknown class label: "197"

Question

我正在尝试使 catboost 模型适合我的数据

#creating the model
model_cb = CatBoostClassifier(task_type='GPU', iterations=100, random_state=2020,eval_metric='F1')
#fitting the dataset into the model
model_cb.fit(X_train, y_train, cat_features= object_cols, plot=True,eval_set=(X_valid, y_valid))

但是当我运行代码时出现错误

CatBoostError: catboost/private/libs/target/target_converter.cpp:228: Unknown class label: "676"

指向文档以了解错误

Answer 1

这是文档https://catboost.ai/docs/concepts/about.html

CatBoostClassifier（二元分类器）期望 y_train 包含浮点数而不是字符串。

请注意，由于我们没有密切监控 Whosebug，因此最好在 github 上提问。com/catboost/catboost

CatBoostError: catboost/private/libs/target/target_converter.cpp:228: Unknown class label: "197"

CatBoostError: catboost/private/libs/target/target_converter.cpp:228: Unknown class label: "197"

machine-learning

catboost

catboostregressor