DeepBaseAbstractModel

class DeepBaseAbstractModel

Bases: abc.ABC

Interface for holding class of etna native deep models.

Inherited-members

Methods

get_model()	Get model.
raw_fit(torch_dataset)	Fit model with torch like Dataset.
raw_predict(torch_dataset)	Make inference on torch like Dataset.

abstract get_model() → etna.models.base.DeepBaseNet

Get model.

Returns

Torch Module

Return type

etna.models.base.DeepBaseNet

abstract raw_fit(torch_dataset: torch.utils.data.dataset.Dataset) → etna.models.base.DeepBaseAbstractModel

Fit model with torch like Dataset.

Parameters

torch_dataset (torch.utils.data.dataset.Dataset) – Samples with data to fit on.

Returns

Trained Model

Return type

etna.models.base.DeepBaseAbstractModel

abstract raw_predict(torch_dataset: torch.utils.data.dataset.Dataset) → Dict[Tuple[str, str], numpy.ndarray]

Make inference on torch like Dataset.

Parameters

torch_dataset (torch.utils.data.dataset.Dataset) – Samples with data to make inference on.

Returns

Predictions for each segment

Return type

Dict[Tuple[str, str], numpy.ndarray]