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

Add support for Chronos ⚡ models #204

Open
wants to merge 24 commits into
base: main
Choose a base branch
from
Open

Add support for Chronos ⚡ models #204

wants to merge 24 commits into from

Conversation

abdulfatir
Copy link
Contributor

@abdulfatir abdulfatir commented Nov 16, 2024

Issue #, if available: N/A

Description of changes: This PR adds support for Chronos ⚡ models.

TODOs:

  • Update evaluation script
  • Fix and add tests for Bolt
  • Update docstrings
  • Update README example and mention Chronos ⚡
  • Update results bar plot in README
  • Add versions for libraries in pyproject.toml
  • Check that the training and eval scripts work
  • Change autogluon -> amazon in model names

NOTE: To be merged after Chronos ⚡ models are available under the amazon namespace on HF.

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

@abdulfatir abdulfatir marked this pull request as draft November 16, 2024 22:31
@abdulfatir abdulfatir changed the title Add support for ChronosBolt models Add support for Chronos ⚡ models Nov 16, 2024
@abdulfatir abdulfatir marked this pull request as ready for review November 17, 2024 19:50
pyproject.toml Show resolved Hide resolved
scripts/evaluation/evaluate.py Show resolved Hide resolved
src/chronos/chronos.py Outdated Show resolved Hide resolved
@@ -0,0 +1,159 @@
# Copyright Amazon.com, Inc. or its affiliates. All Rights Reserved.
# SPDX-License-Identifier: Apache-2.0

Copy link
Contributor

Choose a reason for hiding this comment

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

should we give "links to original source code" to AG in case there is drift?

Copy link
Contributor Author

Choose a reason for hiding this comment

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

Makes sense. Added.

@lostella lostella added the do not merge Should not be merged label Nov 19, 2024
src/chronos/chronos_bolt.py Outdated Show resolved Hide resolved
src/chronos/chronos.py Outdated Show resolved Hide resolved
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment
Labels
do not merge Should not be merged
Projects
None yet
Development

Successfully merging this pull request may close these issues.

3 participants