sub_modules

Implementation of nn.Modules for N-Beats model.

Functions

linear(input_size, output_size[, bias, dropout])
linspace(backcast_length, forecast_length[, …])

Classes

NBEATSBlock(units, thetas_dim[, …])	Initializes internal Module state, shared by both nn.Module and ScriptModule.
NBEATSGenericBlock(units, thetas_dim[, …])	Initializes internal Module state, shared by both nn.Module and ScriptModule.
NBEATSSeasonalBlock(units[, thetas_dim, …])	Initializes internal Module state, shared by both nn.Module and ScriptModule.
NBEATSTrendBlock(units, thetas_dim[, …])	Initializes internal Module state, shared by both nn.Module and ScriptModule.

NBeats pytorch_forecasting.models.nbeats.sub_modules.linear