https://github.com/nixtla logo
r

Rafael Correia Da Silva

05/25/2023, 1:21 PM
hi yall! in neuralforecast/nbeatsx implementation, here especifically, it says that dropout is not implemented yet. maybe im too clueless to tell, but the commented line wouldnt do just fine? thanks for the great lib, having a blast so far using it!
k

Kin Gtz. Olivares

05/25/2023, 1:24 PM
Hey @Rafael Correia Da Silva, Thanks for using NeuralForecast, I just added the dropout to NHITS, and will soon do the same on NBEATSx. Here is a NHITS dropout usage example: https://nixtla.github.io/neuralforecast/examples/robust_regression.html
amazing 1
r

Rafael Correia Da Silva

05/25/2023, 1:24 PM
sweet, thank you!
k

Kin Gtz. Olivares

05/25/2023, 1:46 PM
Thanks for the pointer @Rafael Correia Da Silva, We will merge this https://github.com/Nixtla/neuralforecast/pull/595 into main soon. In the meantime you might be able to work with a dropout regularized NHITS too.
r

Rafael Correia Da Silva

05/25/2023, 1:50 PM
that's amazing, thanks again! it's incredible how fast you guys charge on to tackle feedback.
2 Views