PoolGenerator

class PoolGenerator(configs_template: List[dict])[source]

Bases: object

Generate a pool of pipelines from given config templates in hydra format.

Initialize with a list of config templates in hydra format.

Parameters

configs_template (List[dict]) – list of template configs in hydra format

Notes

Hydra configs templates:

{
    '_target_': 'etna.pipeline.Pipeline',
    'horizon': '${__aux__.horizon}',
    'model': {'_target_': 'etna.models.ProphetModel'}
}

Values to be interpolated should be in the form of ${__aux__.key}

Inherited-members

Parameters

configs_template (List[dict]) –

Methods

generate(horizon)

Fill templates with args.
generate(horizon: int) → List[etna.pipeline.pipeline.Pipeline][source]

Fill templates with args.

Parameters

horizon (int) – horizon to forecast

Return type

List[etna.pipeline.pipeline.Pipeline]