search

LincanLi98 / STG-Mamba

Official Implementation of STG-Mamba: Spatial-Temporal Graph Learning via Selective State Space Model.
134 stars 24 forks source link

seq_len and pred_len #2

Open rginjapan opened 3 months ago

rginjapan commented 3 months ago

Is it possible to use several previous inputs to predict only next one moment? such as seq_len = 12, but pred_len =1, currently the code is only work with seq_len == pred_len

rginjapan commented 3 months ago

However, in the paper you said "We use the historical ground-truth data from the past 12 steps to forecast one future time step", the pred is the same size as input in your implementation, so how and where to get the one future time prediction?

rginjapan commented 3 months ago

How can I change this model to multi-ouputs(1 to k moment) to one output (k+1 moment)? using the average of dim=1 to convert the prediction of (b,12,184) to (b,1,184)??