pchen182224 commited on
Commit
0ce2f09
·
verified ·
1 Parent(s): bb691c6

Upload README.md

Browse files
Files changed (1) hide show
  1. README.md +48 -0
README.md ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+
2
+
3
+ # LightGTS: A Lightweight General Time Series Forecasting Model
4
+
5
+ 🚩 **News (2025.06)** LightGTS has been accepted as **ICML 2025**.
6
+
7
+
8
+
9
+ ## Introduction
10
+
11
+ <div style="text-align: center;">
12
+ <img src="framework.png" alt="LightGTS" style="zoom:80%;" />
13
+ </div>
14
+
15
+
16
+ ## Quick Demos
17
+ ```
18
+ pip install transformers==4.30.2 # Use this version for stable compatibility
19
+ ```
20
+
21
+ ```
22
+ import torch
23
+ from transformers import AutoModelForCausalLM
24
+ # load pretrain model
25
+ # supports different lookback/forecast lengths
26
+ model = AutoModelForCausalLM.from_pretrained('DecisionIntelligence/LightGTS', trust_remote_code=True)
27
+ # prepare input
28
+ batch_size, lookback_length = 1, 528
29
+ seqs = torch.randn(batch_size, lookback_length).unsqueeze(-1).float()
30
+ # Note that Sundial can generate multiple probable predictions
31
+ forecast_length = 192
32
+ outputs = model.generate(seqs, patch_len = 48, stride_len=48, max_output_length=forecast_length, inference_patch_len=48)
33
+ print(output.shape)
34
+ ```
35
+
36
+
37
+ ## Citation
38
+
39
+ If you find Sundial helpful for your research, please cite our paper:
40
+ ```
41
+ @article{wang2025lightgts,
42
+ title={LightGTS: A Lightweight General Time Series Forecasting Model},
43
+ author={Wang, Yihang and Qiu, Yuying and Chen, Peng and Shu, Yang and Rao, Zhongwen and Pan, Lujia and Yang, Bin and Guo, Chenjuan},
44
+ journal={arXiv preprint arXiv:2506.06005},
45
+ year={2025}
46
+ }
47
+
48
+ ```