neuralforecast icon indicating copy to clipboard operation
neuralforecast copied to clipboard

NBEATS forward is returning all-nan forecasts

Open alejandroxag opened this issue 3 years ago • 2 comments

The forward method from the NBEATS model (https://github.com/Nixtla/nixtlats/blob/a3c7442a4c16c255685e158c9347d045f87ffa3b/nixtlats/models/nbeats/nbeats.py#L402) is returning all-nan forecast tensors. The real issue comes from the forecast method (https://github.com/Nixtla/nixtlats/blob/a3c7442a4c16c255685e158c9347d045f87ffa3b/nixtlats/models/nbeats/nbeats.py#L431) when the argument return_decomposition is set to False. (https://github.com/Nixtla/nixtlats/blob/a3c7442a4c16c255685e158c9347d045f87ffa3b/nixtlats/models/nbeats/nbeats.py#L424)

This issue has only been detected when using MASE Loss as the training loss function.

alejandroxag avatar Jul 07 '21 18:07 alejandroxag

@all-contributors please add @alejandroxag for code

mergenthaler avatar Feb 04 '22 17:02 mergenthaler

@mergenthaler

I've put up a pull request to add @alejandroxag! :tada:

allcontributors[bot] avatar Feb 04 '22 17:02 allcontributors[bot]