Skip to content
New issue

Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.

By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.

Already on GitHub? Sign in to your account

DynamicNBEATs model #1191

Open
wants to merge 1 commit into
base: main
Choose a base branch
from
Open

Conversation

tblume1992
Copy link

By default NBEATs only utilizes a polynomial basis for the trend component. This PR adds support for several different basis functions for the trend and adds a handler for these functions in a new 'DynamicNBEATs' method.

Supports several basis functions for the trend and adds a handler for these functions
Copy link

Check out this pull request on  ReviewNB

See visual diffs & provide feedback on Jupyter Notebooks.


Powered by ReviewNB

@CLAassistant
Copy link

CLA assistant check
Thank you for your submission! We really appreciate it. Like many open source projects, we ask that you sign our Contributor License Agreement before we can accept your contribution.
You have signed the CLA already but the status is still pending? Let us recheck it.

@elephaint
Copy link
Contributor

elephaint commented Nov 14, 2024

Hi, thanks for your contribution, this is really cool! Two comments:

  1. Could you please follow the contributing guidelines? Specifically, we need you to build the library so that the correct python files are generated.
  2. Why not adjust NBEATs to support different basis by adding a parameter that allows to choose the basis? I'd assume that would save a lot of code.

@tblume1992
Copy link
Author

  1. hey @elephaint I was chatting with @marcopeix about this awhile back. Unfortunately I don't have time for a PR but he wanted me to be the author of it. All the code was just in a notebook I threw into the PR.

  2. You can definitely do that and that is basically what I did. 95% of this code is just the base NBEATs code. I think though that this change would alter the 'auto' model pretty significantly since we would want to search for the basis functions AND the number of functions. So we may want to think of it as a different model.

Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
None yet
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants