ModelPipelineParamsToTuneMixin

class ModelPipelineParamsToTuneMixin[source]

Bases: object

Mixin for pipelines with model inside with implementation of params_to_tune method.

Inherited-members

Methods

params_to_tune()

Get hyperparameter grid to tune.

Attributes

model

transforms

params_to_tune() → Dict[str, etna.distributions.distributions.BaseDistribution][source]

Get hyperparameter grid to tune.

Parameters for model has prefix “model.”, e.g. “model.alpha”.

Parameters for transforms has prefix “transforms.idx.”, e.g. “transforms.0.mode”.

Returns

Grid with parameters from model and transforms.

Return type

Dict[str, etna.distributions.distributions.BaseDistribution]