gluonts.shell.util module

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

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',
    ]
}
gluonts.shell.util.invoke_with(fn, *args, **kwargs)[source]

Call fn(*args, **kwargs), but only use kwargs that fn actually uses.