mims-harvard / UniTS

A unified multi-task time series model.
https://zitniklab.hms.harvard.edu/projects/UniTS/
MIT License
411 stars 53 forks source link

How to run self supervised pretraining using UniTS on custom data? #11

Open nikhilajoshy opened 6 months ago

gasvn commented 5 months ago

https://github.com/mims-harvard/UniTS/blob/main/Tutorial.md You can refer to this. The major difference is that you need to define your dataset in the config in such way: https://github.com/mims-harvard/UniTS/blob/main/data_provider/multi_task_pretrain.yaml