mirror of
https://github.com/amazon-science/chronos-forecasting
synced 2026-05-22 09:09:44 +00:00
72ab64166c
*Issue #, if available:* N/A *Description of changes:* This PR adds support for Chronos-Bolt models. TODOs: - [x] Update evaluation script - [x] Fix and add tests for Bolt - [x] Update docstrings - [x] Update README example and mention Chronos-Bolt - [x] Update results bar plot in README - [x] Add versions for libraries in `pyproject.toml` - [x] Check that the training and eval scripts work - [x] Change `autogluon` -> `amazon` in model names Post Merge: - [ ] Update Citation style in README, both Github and HuggingFace repos - [ ] Remove note about AutoGluon - [ ] Update READMEs of original Chronos models to refer to Chronos-Bolt NOTE: To be merged after Chronos-Bolt 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. --------- Co-authored-by: Abdul Fatir Ansari <ansarnd@amazon.de> Co-authored-by: Caner Turkmen <turkmen.ac@gmail.com> Co-authored-by: Lorenzo Stella <stellalo@amazon.com> |
||
|---|---|---|
| .. | ||
| chronos-bolt-base-agg-rel-scores.csv | ||
| chronos-bolt-base-in-domain.csv | ||
| chronos-bolt-base-zero-shot.csv | ||
| chronos-bolt-mini-agg-rel-scores.csv | ||
| chronos-bolt-mini-in-domain.csv | ||
| chronos-bolt-mini-zero-shot.csv | ||
| chronos-bolt-small-agg-rel-scores.csv | ||
| chronos-bolt-small-in-domain.csv | ||
| chronos-bolt-small-zero-shot.csv | ||
| chronos-bolt-tiny-agg-rel-scores.csv | ||
| chronos-bolt-tiny-in-domain.csv | ||
| chronos-bolt-tiny-zero-shot.csv | ||
| chronos-t5-base-agg-rel-scores.csv | ||
| chronos-t5-base-in-domain.csv | ||
| chronos-t5-base-zero-shot.csv | ||
| chronos-t5-large-agg-rel-scores.csv | ||
| chronos-t5-large-in-domain.csv | ||
| chronos-t5-large-zero-shot.csv | ||
| chronos-t5-mini-agg-rel-scores.csv | ||
| chronos-t5-mini-in-domain.csv | ||
| chronos-t5-mini-zero-shot.csv | ||
| chronos-t5-small-agg-rel-scores.csv | ||
| chronos-t5-small-in-domain.csv | ||
| chronos-t5-small-zero-shot.csv | ||
| chronos-t5-tiny-agg-rel-scores.csv | ||
| chronos-t5-tiny-in-domain.csv | ||
| chronos-t5-tiny-zero-shot.csv | ||
| seasonal-naive-in-domain.csv | ||
| seasonal-naive-zero-shot.csv | ||