分享到微信

当前解读页的微信分享二维码

用微信扫一扫打开,点击右上角「···」分享给朋友或朋友圈。

手机上可截屏后,在微信扫一扫中从相册识别二维码。

dmlc/xgboost

机器学习

用 Python、R、Java、Scala 或 C++ 在单机或分布式集群上训练快速、准确的梯度提升树模型。

28,926
8,955
C++
Apache-2.0
创建于 12 年前

XGBoost 是一个用于训练梯度提升决策树(GBDT/GBM)的库,这类模型在表格数据上被广泛使用。它的实际卖点是速度与规模:同一套代码实现并行树提升,并据 README 所述可运行于 Kubernetes、Hadoop、SGE、Dask、Spark、PySpark 等分布式环境,处理「超过数十亿样本」的问题(README)。方法本身记录在 2016 年 KDD 论文中(论文)。

工作流中的变化

你不再从零编写提升算法,而是定义数据和训练参数,由 XGBoost 完成模型拟合,并处理树构建与并行化。库面向多种语言——Python、R、Java、Scala、C 等——团队可以留在现有技术栈中,不必统一为单一语言(仓库)。安装、参数与调优内容不在简短 README 中,应查阅文档确认所选语言和平台的具体设置与 API。

融入现有流水线

由于同一代码据称可同时在本地和分布式环境运行,它既能用于单机实验,也能用于集群任务,无需更换库。R 和 Python 包分别通过各自的包渠道发布,README 中的 CRAN 与 PyPI 徽章反映了这一点(README)。至于某个具体集成(Spark、Dask、Flink 等)是否适配你的集群版本和部署模式,需要查看该集成的文档;README 只在高层面列出支持的环境。

许可

XGBoost 采用 Apache-2.0 许可(LICENSE),允许在该许可条款下商用和修改;README 也将其概括为 Apache-2(README)。

实际考量

README 未说明硬件要求、各平台兼容矩阵或已知故障模式,这些内容应见文档与发布说明(发布说明)。上线前尤其要查阅分布式后端的相关文档。讨论热度不是质量信号:2023 年的 Hacker News 帖子宣布 XGBoost 2.0,但仅获得少量分数和 1 条评论(HN 帖子)。目前唯一一条趋势记录是 2026-10-30 的 28,926 星、当日 +4;单日数据不足以说明趋势。

本文由 AI 根据项目 README 生成,更新于 2026年10月30日,以项目官方文档为准。

Hacker News 讨论

XGBoost2.0
▲ 4 · 1 条评论 · 2023年10月9日
XGBoost v1.0.0 Released
▲ 3 · 0 条评论 · 2020年2月21日
XGBoost 1.0.0 Stable
▲ 2 · 0 条评论 · 2020年3月15日