mirror of
https://github.com/amazon-science/chronos-forecasting
synced 2026-05-24 01:58:27 +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>
1.2 KiB
1.2 KiB
| 1 | dataset | model | MASE | WQL |
|---|---|---|---|---|
| 2 | electricity_15min | amazon/chronos-t5-mini | 0.4446629660227641 | 0.08114657599239496 |
| 3 | m4_daily | amazon/chronos-t5-mini | 3.1533349226194005 | 0.022000507013584743 |
| 4 | m4_hourly | amazon/chronos-t5-mini | 0.7616830292996938 | 0.024630575107847653 |
| 5 | m4_monthly | amazon/chronos-t5-mini | 0.9934074425853089 | 0.10402168689068064 |
| 6 | m4_weekly | amazon/chronos-t5-mini | 2.1407189608104416 | 0.04138058102434373 |
| 7 | monash_electricity_hourly | amazon/chronos-t5-mini | 1.3698378948313894 | 0.09189698159081384 |
| 8 | monash_electricity_weekly | amazon/chronos-t5-mini | 1.9238345295706893 | 0.07015383787479901 |
| 9 | monash_kdd_cup_2018 | amazon/chronos-t5-mini | 0.6027861468526459 | 0.25493489598663444 |
| 10 | monash_london_smart_meters | amazon/chronos-t5-mini | 0.8570035850603943 | 0.4356582737588471 |
| 11 | monash_pedestrian_counts | amazon/chronos-t5-mini | 0.30374539593979855 | 0.2374083216051065 |
| 12 | monash_rideshare | amazon/chronos-t5-mini | 0.8157349455509949 | 0.12963515638823117 |
| 13 | monash_temperature_rain | amazon/chronos-t5-mini | 1.010161905102516 | 0.6919171702485583 |
| 14 | taxi_30min | amazon/chronos-t5-mini | 0.9318379552979712 | 0.31229508015999674 |
| 15 | uber_tlc_daily | amazon/chronos-t5-mini | 0.9213437323817685 | 0.10475291429149586 |
| 16 | uber_tlc_hourly | amazon/chronos-t5-mini | 0.6812621470377416 | 0.15982192635434303 |