lightgbm.LGBMModel 和 lightgbm.LGBMClassifier 有什么区别?

数据挖掘 Python scikit-学习
2022-03-01 04:55:56

https://lightgbm.readthedocs.io/en/latest/Python-API.html

我在 lightgbm 文档中比较了它们,但不知道我会选择哪一个,为什么我会选择另一个?

1个回答

查看源代码(可从您的链接获得),它似乎是(以及 Ranker 和 Regressor)LGBMModel的父类。LGBMClassifier您可能应该坚持使用分类器;它强制执行适当的损失函数,添加一组数据类,将模型的分数转换为类概率,并从那里转换为预测类等。