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.

Warning

This method uses dill module which is not secure. It is possible to construct malicious data which will execute arbitrary code during loading. Never load data that could have come from an untrusted source, or that could have been tampered with.

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.