轻量级GBM - Python API vs Scikit-learn API

Question

9

我正在尝试使用LightGBM，并已经阅读了Python API 文档。

训练API和Scikit-learn API之间有什么区别吗？我们是否可以同时使用这两个API来解决同一个问题并得到相同的结果？

- DIPANJAN MITRA

1个回答

网页内容由stack overflow 提供, 点击上面的

可以查看英文原文，
原文链接

- Mischa Lisovyi · Accepted Answer

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

原因是sklearn API只是“本地训练”API的封装，而后者又是后端C++库的封装。最终，这是你自己的选择。我个人建议使用sklearn API，有两个主要优点：

您可以利用完整的sklearn工具包(包括数据预处理、超参数优化、模型评估等流水线)
您可以无痛地在不同模型之间切换，即您的输入数据具有相同的格式（pd.DataFrame或np.ndarray），训练接口相同，并且您可以通过实例化不同的对象并将它们通过相同的过程传递来在sklearn模型、lightgbm、xgboost、catboost或vowpal wabbit之间切换。