时间序列框架Darts踩坑记录

Darts是一个时间序列分析框架，支持多种时间序列模型，GitHub (opens new window)。虽然提供了很多模型和方法，该框架的文档还是存在较多的缺失，很多时候需要到源代码里自行查找，故在此总结一些文档里没有提到的但是踩过的坑。

自定义模型

自定义模型初始化参数

class PLForecastingModule(pl.LightningModule, ABC):
    @abstractmethod
    def __init__(
        self,
        input_chunk_length: int,
        output_chunk_length: int,
        loss_fn: nn.modules.loss._Loss = nn.MSELoss(),
        likelihood: Optional[Likelihood] = None,
        optimizer_cls: torch.optim.Optimizer = torch.optim.Adam,
        optimizer_kwargs: Optional[Dict] = None,
        lr_scheduler_cls: torch.optim.lr_scheduler._LRScheduler = None,
        lr_scheduler_kwargs: Optional[Dict] = None,
    ) -> None: