0

我正在尝试将 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"

将我指向文档以了解错误

4

1 回答 1

2

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

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

请注意，由于我们不会非常密切地监控 stackoverflow，因此最好在 github.com/catboost/catboost 提问

于 2020-08-28T05:39:57.300 回答