python - 在 Catboost 中使用具有多类的 GPU

Question

我正在尝试使用 catboost 来预测多类，但出现如下错误。

Error: loss function is not supported for GPU learning MultiClass

我使用了 catboost 选项，如下所示。

CatboostClassifier(loss_function='MultiClass',task_type='GPU') 

我想知道我是否可以使用 GPU 来预测多分类。

标签： pythoncatboost

Answer 1

从 0.10.0 版开始，MultiClass 可用于 GPU catboost。