https://github.com/nixtla logo
Title
c

clever-telephone-90805

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!
s

stale-businessperson-86432

05/25/2023, 1:24 PM
Hey @clever-telephone-90805, 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
c

clever-telephone-90805

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

stale-businessperson-86432

05/25/2023, 1:46 PM
Thanks for the pointer @clever-telephone-90805, 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.
c

clever-telephone-90805

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