light gbm - python API 与 Scikit-learn API

DIP*_*TRA 6 python scikit-learn lightgbm

我正在尝试应用 LightGBM 并浏览了 Python API文档

Training API 和 Scikit-learn API 之间有什么区别吗?我们可以使用这两个 API 来针对同一问题实现相同的结果吗?

Myk*_*vyi 7

简短的回答:是的,如果您以相同的方式配置它们,它们将提供相同的结果。

原因是 sklearn API 只是“本机训练”API 的包装器,而后者又是后端 C++ 库的包装器。最后,这是您的选择。我个人会建议使用 sklearn API。2个主要优点是:

  • 您可以使用完整的 sklearn 工具包(具有数据预处理、超参数优化、模型评估等的管道)
  • 您可以轻松地在不同模型之间切换,即您的输入数据具有相同的格式(pd.DataFramenp.ndarray),trainign 界面相同,您可以通过简单地实例化不同的对象和通过相同的程序传递它们。