使用XGBoost磚

學習如何在磚使用XGBoost訓練機器學習模型。磚運行時機器學習包括XGBoost Python和Scala庫。

火車XGBoost模型在單個節點上

你可以訓練模型使用Pythonxgboost包中。這個包隻支持單一節點的工作負載。訓練PySpark毫升管道和利用分布式訓練,明白了分布式XGBoost模型的訓練。

分布式訓練XGBoost模型、磚包括PySpark估計量的基礎上xgboost包中。磚還包括Scala的包xgboost-4j。對於細節和示例的筆記本,見以下: