7

我正在尝试应用 LightGBM 并阅读了 Python API文档

Training API 和 Scikit-learn API 之间有什么区别吗?我们可以同时使用这两种 API 来解决相同的问题吗?

4

1 回答 1

9

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

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

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