Skip to main content
Wrapper of synapse.ml.lightgbm.LightGBMRegressor that adds an extract_local_model method to get a local version of the trained model and broadcast it to the workers.

SparkLGBMForecast

Bases: LightGBMRegressor

SparkLGBMForecast.extract_local_model

extract_local_model(trained_model)