search

tsinghua-fib-lab / UniST

Official implementation for "UniST: A Prompt-Empowered Universal Model for Urban Spatio-Temporal Prediction" (KDD 2024)
147 stars 14 forks source link

Prediction length problem #20

Open liyuzhan opened 2 months ago

liyuzhan commented 2 months ago

Hello author, I want to change the length of pred_len from 6 to 1, but when running the code, the following problem appears, how to solve it? Traceback (most recent call last): File "/data3/UniST-main/src/main.py", line 164, in main() File "/data3/UniST-main/src/main.py", line 113, in main assert args.his_len + args.pred_len == args.seq_len AssertionError

YuanYuan98 commented 2 months ago

There are three parameters that define the prediction task: his_len, pred_len, and seq_len. Please ensure that the sum of pred_len and his_len equals seq_len to avoid this assertion error.