Update README.md
Browse files
README.md
CHANGED
@@ -8,9 +8,9 @@ pipeline_tag: time-series-forecasting
|
|
8 |
|
9 |
Large time-series model introduced in this [paper](https://arxiv.org/abs/2402.02368) and enhanced with our [further work](https://arxiv.org/abs/2410.04803).
|
10 |
|
11 |
-
This version is pre-trained on **307B** time points,
|
12 |
-
|
13 |
-
|
14 |
|
15 |
# Quickstart
|
16 |
|
@@ -28,7 +28,9 @@ normed_seqs = (seqs - mean) / std
|
|
28 |
|
29 |
# forecast
|
30 |
prediction_length = 96
|
31 |
-
|
|
|
|
|
32 |
print(output.shape)
|
33 |
```
|
34 |
|
|
|
8 |
|
9 |
Large time-series model introduced in this [paper](https://arxiv.org/abs/2402.02368) and enhanced with our [further work](https://arxiv.org/abs/2410.04803).
|
10 |
|
11 |
+
This version is pre-trained on **307B** time points with **84M** parameters, a lightweight generative Transformer with the state-of-the-art performance on zero-shot forecasting:
|
12 |
+
|
13 |
+
We evaluate the model on the following benchmarks: [TSLib Dataset](), [GIFT-Eval]().
|
14 |
|
15 |
# Quickstart
|
16 |
|
|
|
28 |
|
29 |
# forecast
|
30 |
prediction_length = 96
|
31 |
+
normed_output = model.generate(normed_seqs, max_new_tokens=prediction_length)[:, -prediction_length:]
|
32 |
+
output = std * normed_output + mean
|
33 |
+
|
34 |
print(output.shape)
|
35 |
```
|
36 |
|