SaveModelPipelineMixin

class SaveModelPipelineMixin

Bases: etna.core.mixins.SaveMixin

Implementation of AbstractSaveable abstract class for pipelines with model inside.

It saves object to the zip archive with 4 entities:

• metadata.json: contains library version and class name.

• object.pkl: pickled without model, transforms and ts.

• model.zip: saved model.

• transforms: folder with saved transforms.

Inherited-members

Methods

load(path[, ts])	Load an object.
save(path)	Save the object.

Attributes

model
transforms
ts

classmethod load(path: pathlib.Path, ts: Optional[etna.datasets.tsdataset.TSDataset] = None) → typing_extensions.Self

Load an object.

Parameters

• path (pathlib.Path) – Path to load object from.

• ts (Optional[etna.datasets.tsdataset.TSDataset]) – TSDataset to set into loaded pipeline.

Returns

Loaded object.

Return type

typing_extensions.Self

save(path: pathlib.Path)

Save the object.

Parameters

path (pathlib.Path) – Path to save object to.