在Catboost中将GPU与多类一起使用

时间:2018-09-04 06:13:16

标签: python catboost

我正在尝试使用catboost预测多类,但出现以下错误。

Error: loss function is not supported for GPU learning MultiClass

我使用了如下的catboost选项。

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

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

1 个答案:

答案 0 :(得分:1)

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