base_model

Timeseries models share a number of common characteristics. This module implements these in a common base class.

Functions

_concatenate_output(output)	Concatenate multiple batches of output dictionary.
_torch_cat_na(x)	Concatenate tensor along dim=0 and add nans along dim=1 if necessary.

Classes

AutoRegressiveBaseModel(log_interval, …[, …])	Model with additional methods for autoregressive models.
AutoRegressiveBaseModelWithCovariates(…[, …])	Model with additional methods for autoregressive models with covariates.
BaseModel(log_interval, float] = -1, …[, …])	BaseModel from which new timeseries models should inherit from.
BaseModelWithCovariates(log_interval, …[, …])	Model with additional methods using covariates.

models pytorch_forecasting.models.base_model._concatenate_output