unit8co / darts

A python library for user-friendly forecasting and anomaly detection on time series.

https://unit8co.github.io/darts/

Apache License 2.0

7.92k stars 859 forks source link

[BUG] Finish TODOs in NHiTs and NBEATS #954

Closed gdevos010 closed 2 years ago

gdevos010 commented 2 years ago

Describe the bug Make NHiTs activation configurable. https://github.com/unit8co/darts/blob/5213854b71e3d56d41dcdffac59140a8da9c10a7/darts/models/forecasting/nhits.py#L86

Apply NHiTs dropout and batch_norm to NBEATs https://github.com/unit8co/darts/blob/5213854b71e3d56d41dcdffac59140a8da9c10a7/darts/models/forecasting/nhits.py#L130

NHiTs Support for AvgPooling1d https://github.com/unit8co/darts/blob/5213854b71e3d56d41dcdffac59140a8da9c10a7/darts/models/forecasting/nhits.py#L108

gdevos010 commented 2 years ago

955