我正在尝试将CatBoost模型拟合到我的数据上
#creating the modelmodel_cb = CatBoostClassifier(task_type='GPU', iterations=100, random_state=2020,eval_metric='F1')#fitting the dataset into the modelmodel_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: 未知类别标签: "676"
请指引我到文档以了解这个错误
回答:
这是文档 https://catboost.ai/docs/concepts/about.html
CatBoostClassifier(一个二元分类器)期望y_train包含浮点数而不是字符串。
请注意,由于我们不太密切关注stackoverflow,最好在github.com/catboost/catboost上提问