thuml
/

timer-base-84m

Time Series Forecasting

foundation-models

pretrained-models

time-series-foundation-models

large-time-series-models

Model card Files Files and versions Community

Yong99 commited on 27 days ago

Commit

0257255

·

verified ·

1 Parent(s): 0b4e50a

Update prediction_example_etth1.ipynb

Files changed (1) hide show

prediction_example_etth1.ipynb +1 -1

prediction_example_etth1.ipynb CHANGED Viewed

@@ -44,7 +44,7 @@
    "source": [
     "# generate forecast \n",
     "prediction_length = 288     # forecast the next 288 timestamps, supporting maximum prediction length encompassed in the context length\n",
-    "normed_output = model.generate(normed_seqs, max_new_tokens=prediction_length)[:, -prediction_length:]\n",
     "output = std * normed_output + mean                                             # rescale the output to the original scale"
    ]
   },

    "source": [
     "# generate forecast \n",
     "prediction_length = 288     # forecast the next 288 timestamps, supporting maximum prediction length encompassed in the context length\n",
+    "normed_output = model.generate(normed_seqs, max_new_tokens=prediction_length)\n",
     "output = std * normed_output + mean                                             # rescale the output to the original scale"
    ]
   },