unit8co/darts

Write an `NBEATSExplainer` class for the interpretable version

Open

#1,270 opened on Oct 9, 2022

View on GitHub
 (1 comment) (0 reactions) (0 assignees)Python (6,832 stars) (762 forks)batch import
good first issueimprovement

Description

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.

Contributor guide

Tech stack
python
Domain
machine learningdata
Issue type
feature
DifficultyEstimated implementation difficulty for a new contributor, from 1 for very small changes to 5 for expert-level work.
3
Estimated timeA rough time range for an experienced contributor to investigate, implement, test, and prepare a pull request.
1-3 hours
Activity statusHow available the issue appears right now: fresh, active, stale, blocked, or waiting on maintainer input.
fresh
ClarityHow clearly the issue explains the expected change, acceptance criteria, and next step.
clear
Prerequisites
Basic knowledge of NBEATS modelFamiliarity with darts library
Newbie friendlinessA 1-100 score estimating how approachable this issue is for first-time contributors.
50
Research direction
Examine the existing NBEATS implementation in darts (likely in darts.models.forecasting.nbeats) to understand how the interpretable mode decomposes forecasts into trend and seasonality components. Then, create a new class NBEATSExplainer in darts.explainability that exposes these components, following the patterns used by other explainer classes in the module. Consider reviewing any related discussions in issue comments for additional context.
Write an `NBEATSExplainer` class for the interpretable version · unit8co/darts#1270 | Good First Issue