N-BEATS icon indicating copy to clipboard operation
N-BEATS copied to clipboard

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

Open PhoebeChen123 opened this issue 3 years ago • 0 comments

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?

PhoebeChen123 avatar Dec 15 '21 03:12 PhoebeChen123