Is

dropout

for

NBEATS

currently under construction? I get the following message "_NotImplementedError: dropout_" when trying to use it currently. Thanks for the clarification!

The parameter is called

dropout_prob

for NBEATS.

Thanks for the reply Marco! From my review and try-outs it seems it is implemented for NBEATSx though not for NBEATS. In any case, using

NBEATSx

with the dropout parameter

dropout_prob_theta

works perfectly 🙌