lightgbm.LGBMModel和lightgbm.LGBMClassifier该用哪个?

  统计/机器学习 监督式学习 Python    浏览次数:15576        分享
1

python的lightgbm模块里面好几个模型,一个是lightgbm.LGBMClassifier,还有一个是lightgbm.LGBMRegressor,此外还有一个是lightgbm.LGBMModel。我现在是要做分类,到底是用LGBMClassifier还是LGBMModel?这两个效果有什么区别吗?

 

jimmy   2018-11-05 13:47



   1个回答 
4

直接用的话,分类是LGBMClassifier,回归是LGBMRegressor。

如果要用LGBMModel,要在参数objective指定你的任务类型(regression, binary, multiclass, lambdarank),我一般都是直接用LGBMClassifier或者LGBMRegressor,没用过LGBMModel,不过我觉得指定了任务类型之后,其他参数一样的情况下,效果应该是一样的,因为LGBMClassifier是继承自LGBMModel,objective默认设置为binary或者multiclass。

SofaSofa数据科学社区DS面试题库 DS面经

nlceyes   2018-11-05 20:35



  相关讨论

sklearn里的RandomForestClassifier怎么选样本的数量

怎么获得sklearn里决策树的实际深度?

sklearn里的RandomForestClassifier和ExtraTreesClassifier有什么区别吗?

对sklearn的决策树进行剪枝

sklearn随机森林设置每棵树的权重

sklearn DecisionTreeRegressor模型中为什么会有随机种子random_state?

sklearn模块里的DecisionTreeClassifier为什么每次结果不同(数据相同的情况下)

怎么得到sklearn里的RandomForestClassifier训练好之后每棵树的叶节点的个数?

sklearn里怎么获得adaboost中每棵树的权重?

可视化sklearn决策树的结构

  随便看看

为什么梯度的反方向是函数下降最快的方向?

样本权重大小不同时,用什么分类算法比较合适?

查看sklearn的线性回归系数的p value

神经网络中的dense和sparse是什么意思?

两个凸函数相加,还是凸函数吗?