Write an `NBEATSExplainer` class for the interpretable version

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.87k stars 851 forks source link

Write an `NBEATSExplainer` class for the interpretable version #1270

Open hrzn opened 1 year ago

hrzn commented 1 year ago

NBEATS has an "interpretable" mode where it decomposes the forecast in trend and seasonality. We should have a class NBEATSExplainer in darts.explainability which exposes these.

jacke9693 commented 1 year ago

How do I go about accessing them in the meantime, the trend and seasonality components? I've tried looking in the documentation but they seem to be closer to the nn.Module and not so exposed. Is there an easy way to reach them through the API?