The shape of dataloader output y is [batch_num, n_route]?

[Yanan Li]

It seems you only predict one future slot, which is not n_pred.

[Guanyi Yu]

Hi, @YananLi18
Yes, this code only predicts one slot, and n_pred is the time offset of the prediction.

[Yanan Li]

Thanks for your quick answer @Aguin.
I found that the original author also wrote this way, which is somewhat different from the general prediction problem (generally n_pred future slots).