search

ServiceNow / N-BEATS

N-BEATS is a neural-network based model for univariate timeseries forecasting. N-BEATS is a ServiceNow Research project that was started at Element AI.
Other
508 stars 116 forks source link

why does forward of GenericBasis Class in models/nbeats.py return theta instead of FC projection of theta? #14

Open PhoebeChen123 opened 2 years ago

PhoebeChen123 commented 2 years ago

class GenericBasis(t.nn.Module): """ Generic basis function. """ def init(self, backcast_size: int, forecast_size: int): super().init() self.backcast_size = backcast_size self.forecast_size = forecast_size

def forward(self, theta: t.Tensor):
    return theta[:, :self.backcast_size], theta[:, -self.forecast_size:]

is it more reasonable to return a function of theta just like trendBasis and seasonalityBasis?