DeepBaseNet

class DeepBaseNet[source]

Bases: etna.models.base.DeepAbstractNet, pytorch_lightning.core.module.LightningModule

Class for partially implemented LightningModule interface.

Init DeepBaseNet.

Methods

training_step(batch, *args, **kwargs)

Training step.

validation_step(batch, *args, **kwargs)

Validate step.

Attributes

training_step(batch: dict, *args, **kwargs)[source]

Training step.

Parameters

batch (dict) – batch of data

Returns

loss

validation_step(batch: dict, *args, **kwargs)[source]

Validate step.

Parameters

batch (dict) – batch of data

Returns

loss