gluonts.shell.util module

gluonts.shell.util.forecaster_type_by_name(name: str) → Type[Union[gluonts.model.estimator.Estimator, gluonts.model.predictor.Predictor]]

Loads a forecaster from the gluonts_forecasters entry_points namespace by name.

If a forecater wasn’t register under that name, it tries to locate the class.

Third-party libraries can register their forecasters as follows by defining a corresponding section in the entry_points section of their setup.py:

entry_points={
    'gluonts_forecasters': [
        'model_a = my_models.model_a:MyEstimator',
        'model_b = my_models.model_b:MyPredictor',
    ]
}