sub_modules#

Functions

init_weights(module, initialization)

Classes

IdentityBasis(backcast_size, forecast_size, ...)

Initializes internal Module state, shared by both nn.Module and ScriptModule.

NHiTS(context_length, prediction_length, ...)

N-HiTS Model.

NHiTSBlock(context_length, ...)

N-HiTS block which takes a basis function as an argument.

StaticFeaturesEncoder(in_features, out_features)

Initializes internal Module state, shared by both nn.Module and ScriptModule.