使用XGBoost砖
学习如何在砖使用XGBoost训练机器学习模型。砖运行时机器学习包括XGBoost Python和Scala库。
火车XGBoost模型在单个节点上
你可以训练模型使用Pythonxgboost
包中。这个包只支持单一节点的工作负载。训练PySpark毫升管道和利用分布式训练,明白了分布式XGBoost模型的训练。
分布式XGBoost模型的训练
分布式训练XGBoost模型、砖包括PySpark估计量的基础上xgboost
包中。砖还包括Scala的包xgboost-4j
。对于细节和示例的笔记本,见以下:
分布式训练使用xgboost.spark XGBoost模型(砖运行时的12.0毫升以上)
分布式训练使用sparkdl.xgboost XGBoost模型(弃用砖开始运行时12.0毫升)