Data

Tools

Indexes

Applications

Experiments

Open Source Science

Recent Releases of uni2ts

uni2ts - 1.2.0

Version 1.2.0

We are happy to announce the Uni2TS 1.2.0 release. In this version, we provide the code for Moirai-MoE which is our latest model. The model weights can be downloaded from huggingface: small and base.

Besides Moirai-MoE, we also provide more evaluation files to facilite the evaluation for some foundation models (e.g., Chronos, TimesFM, VisionTS). Meanwhile, of course, some enhancements and fixes (e.g., save huggingface checkpoint, training resume) are contained in this version :)

What's Changed in the Codebase

Moirai-MoE

  • update for moirai-moe by @liuxu77 in https://github.com/SalesforceAIResearch/uni2ts/pull/139
  • Update readme file by @liuxu77 in https://github.com/SalesforceAIResearch/uni2ts/pull/140
  • update readme and notebooks for moe by @liuxu77 in https://github.com/SalesforceAIResearch/uni2ts/pull/142

More Evaluation files

  • Evaluate chronos on monash by @liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/72
  • update evaluation code and scripts by @liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/113
  • Scripts and code to evaluate timesfm by @liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/119
  • add VisionTS scripts and code by @Keytoyze in https://github.com/SalesforceAIResearch/uni2ts/pull/120
  • fix VisionTS context len issue by @Keytoyze in https://github.com/SalesforceAIResearch/uni2ts/pull/124

Enhancements and Fixes

  • Add ckpt_path config to resume training by @gorold in https://github.com/SalesforceAIResearch/uni2ts/pull/67
  • Documentation by @gorold in https://github.com/SalesforceAIResearch/uni2ts/pull/71
  • Change to python3 for black in pre-commit by @liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/73
  • add config file for lightning checkpoint by @gorold in https://github.com/SalesforceAIResearch/uni2ts/pull/96
  • change the version name and the project name for pypi by @liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/106
  • Implement a new callback HugginfaceCheckpoint by @liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/109
  • Manually set 'freq' for specific datasets by @zqiao11 in https://github.com/SalesforceAIResearch/uni2ts/pull/110
  • Update README and config files with Moirai-1.1-R and fix Moirai-1.0-R model weights by @gorold in https://github.com/SalesforceAIResearch/uni2ts/pull/111
  • update ipynb examples with 1.1 models by @liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/112
  • update readme by @liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/131
  • update README by @liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/152
  • update version by @liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/153
  • Update init.py adding version by @liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/155
  • update workflows by @liu-jc in https://github.com/SalesforceAIResearch/uni2ts/pull/157, https://github.com/SalesforceAIResearch/uni2ts/pull/158

New Contributors

  • @liu-jc made their first contribution in https://github.com/SalesforceAIResearch/uni2ts/pull/72
  • @zqiao11 made their first contribution in https://github.com/SalesforceAIResearch/uni2ts/pull/110
  • @Keytoyze made their first contribution in https://github.com/SalesforceAIResearch/uni2ts/pull/120
  • @liuxu77 made their first contribution in https://github.com/SalesforceAIResearch/uni2ts/pull/139

Full Changelog: https://github.com/SalesforceAIResearch/uni2ts/compare/1.1.0...1.2.0

- Jupyter Notebook
Published by liu-jc over 1 year ago

uni2ts - 1.1.0

New features

Add hparams context manager, allowing for adjusting hparams of MoiraiForecast without reloading the model (#20) by @gorold Update MoiraiModule to subclass PyTorchModelHubMixin, enabling download stats tracking (#39) by @gorold

Fixes

Fix bug in MultiOutSizeLinear which masked certain outputs, impacting predictions of patch size 8 and 16 in the current pre-trained model (#46) by @gorold Fix the handling of offset for long dataframes (#48) by @gorold Fix mask_length argument for MoiraiFinetune (#49) by @gorold

Other Changes

Updated cli.eval script to be more similar with training script (#4) by @gorold Added a Project Page for Moirai-1.0-R with instructions to reproduce paper results (#17) by @gorold Update example notebooks with instructions for forecasting with exogenous variables (#35) by @chenghaoliu89

- Jupyter Notebook
Published by chenghaoliu89 almost 2 years ago