Skip to main content
Wrapper of xgboost.spark.SparkXGBRegressor that adds an extract_local_model method to get a local version of the trained model and broadcast it to the workers.

SparkXGBForecast

Bases: SparkXGBRegressor

SparkXGBForecast.extract_local_model

extract_local_model(trained_model)