使用 Scala 对 XGBoost 模型进行分布式训练

示例笔记本演示如何将 XGBoost 与 MLlib 配合使用:

  • 第一个示例演示如何将 XGBoost 模型嵌入 MLlib ML 管道。
  • 第二个示例演示如何使用 MLlib 交叉验证来优化 XGBoost 模型。

使用 ML 管道笔记本实现 XGBoost 分类

获取笔记本

使用交叉验证笔记本实现 XGBoost 回归

获取笔记本