LightGBM的多分类

问题描述

我正在使用最新版本的LightGBM解决多分类问题。当我将目标切换为“多类”时，会发生此错误；

Error in data$update_params(params) : 
  [LightGBM] [Fatal] Number of classes should be specified and greater than 1 for multiclass training

我留下了一个可重复的例子来说明我的方式

catnames <- names(purrr::keep(train_x,is.factor))

dtrain <- lgb.Dataset(as.matrix(train_x),label = train_y,categorical_feature = catnames)
data_file <- tempfile(fileext = ".data")
lgb.Dataset.save(dtrain,data_file)
dtrain <- lgb.Dataset(data_file)
lgb.Dataset.construct(dtrain)

model <- lgb.train(data=dtrain,objective = "multiclass",alpha = 0.1,nrounds = 1000,learning_rate = .1
                   )

试图将我的目标（train_y）保存为因子，但未做任何更改。

解决方法

在LightGBM中使用多类别目标时，您需要传递另一个参数，该参数告诉学习者要预测的类别数量。

因此，它看起来应该更像这样：

model <- lgb.train(data=dtrain,objective = "multiclass",num_classes = INSERT NUMBER OF TARGET CLASSES HERE,alpha = 0.1,nrounds = 1000,learning_rate = .1,)

我对python API的了解更多，所以可能是（如果这不起作用）您需要以lgb.train中的params关键字参数的列表形式传递num_class参数。