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

[timeseries] Add support for categorical covariates #4037

Merged
merged 7 commits into from
Apr 5, 2024

Conversation

shchur
Copy link
Collaborator

@shchur shchur commented Apr 4, 2024

Description of changes:

  • Add support for categorical covariates for all MLForecast + GluonTS models
    • For GluonTS models that only support real-valued covariates, we one-hot-encode the categorical covariates

To do:

  • Extend tests for GluonTS models with different covariates combinations

By submitting this pull request, I confirm that you can use, modify, copy, and redistribute this contribution, under the terms of your choice.

@shchur shchur requested a review from canerturkmen April 4, 2024 15:09
@yinweisu
Copy link
Collaborator

yinweisu commented Apr 4, 2024

Previous CI Run Current CI Run
pydantic_core==2.16.3 -
ray==2.10.0 ray==2.6.3
pydantic==2.6.4 pydantic==1.10.15
annotated-types==0.6.0 -
platformdirs==4.2.0 platformdirs==3.11.0
virtualenv==20.25.1 virtualenv==20.21.0
ray==2.10.0 ray==2.6.3
- gpustat==1.1.1
pydantic==2.6.4 pydantic==1.10.15
- blessed==1.20.0
- nvidia-ml-py==12.535.133
platformdirs==4.2.0 platformdirs==3.11.0
virtualenv==20.25.1 virtualenv==20.21.0

@yinweisu
Copy link
Collaborator

yinweisu commented Apr 4, 2024

Previous CI Run Current CI Run

Copy link
Contributor

@canerturkmen canerturkmen left a comment

Choose a reason for hiding this comment

The reason will be displayed to describe this comment to others. Learn more.

LGTM overall! Just dropped a few comments.

@yinweisu
Copy link
Collaborator

yinweisu commented Apr 5, 2024

Previous CI Run Current CI Run

@shchur shchur merged commit 4de3056 into autogluon:master Apr 5, 2024
28 checks passed
@shchur shchur deleted the cat-covariates branch April 5, 2024 11:59
Copy link

github-actions bot commented Apr 5, 2024

Job PR-4037-7880876 is done.
Docs are uploaded to http://autogluon-staging.s3-website-us-west-2.amazonaws.com/PR-4037/7880876/index.html

@shchur shchur mentioned this pull request Apr 5, 2024
23 tasks
LennartPurucker pushed a commit to LennartPurucker/autogluon that referenced this pull request Jun 1, 2024
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.

None yet

3 participants