Supported target types are: ('binary', 'multiclass'). Got 'continuous' instead.

最新推荐文章于 2023-10-31 10:55:05 发布

sigma2beta

最新推荐文章于 2023-10-31 10:55:05 发布

阅读量5k

点赞数 2

分类专栏：问题指南文章标签：调参 lightgbm

本文链接：https://blog.csdn.net/hiahia28/article/details/103439488

版权

问题指南专栏收录该内容

0 篇文章 0 订阅

订阅专栏

用lgb.cv对回归问题进行调参，会出现如下问题：

Supported target types are: ('binary', 'multiclass'). Got 'continuous' instead.

原因如下，在lgb.cv中的stratify参数默认为True。但是分层（StratifiedKFold）仅适用于分类问题。因此，要使用回归则需要将该参数的值设置为False。

cv_results = lgb.cv(params,
                                lgb_train,
                                seed=2018,
                                nfold=3,
                                metrics=['rmse'],
                                early_stopping_rounds=450,
                                verbose_eval=True,
                                stratified=False
                              )

如上即可正常运行了。