light gbm - python API 与 Scikit-learn API

Question

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

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

Answer 1

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

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

• 您可以使用完整的 sklearn 工具包（具有数据预处理、超参数优化、模型评估等的管道）
• 您可以轻松地在不同模型之间切换，即您的输入数据具有相同的格式（ pd.DataFrame或np.ndarray ），trainign 接口是相同的，您可以在 sklearn 模型、lightgbm、xgboost、catboost 或 vawpal wabbit 之间切换通过简单地实例化不同的对象并将它们传递给相同的过程。