PytorchForecastingMixin

class PytorchForecastingMixin

Bases: object

Mixin for Pytorch Forecasting models.

Inherited-members

Methods

fit(ts)

Fit model.

Attributes

trainer_params
dataset_builder
train_batch_size
test_batch_size
encoder_length

fit(ts: etna.datasets.tsdataset.TSDataset)

Fit model.

Parameters

ts (etna.datasets.tsdataset.TSDataset) – TSDataset to fit.

Returns

model