Why output dimension is only select the first?

[kingofkill]

The code metrics = engine.train(trainx, trainy[:, 0, :, :]) in line84 of train.py seems only predict one dimension of total D dimension.
But the paper wrote the output dimension is D.

[nataliekoh]

I have the same question.

[boykovdn]

Output dimension is 12, which is the number of steps ahead their model predicts. The shape of trainy is (B,2,N,T) where B is the batch size, N is the number of nodes, and T is 12. The second dimension here is the features of the data as they come in from the loader, which is the traffic flow (z-normalised) at 0, and the periodic signal at 1. They only want to predict the traffic flow, so they take the 0th dimension.

[HQ-LV]

Hello! If I want the feature_dimension in the output of gwnet in model.py to be 2 (the default feature_dimension=1 in the source code), how should I modify the gwnet part?
Additionally, I have a question: I understand that the convolution operations in gwnet are performed along the feature dimension, and the output shape of the model is [batch_size, feature_dim, num_nodes, seq_len] where feature_dimension=12 and seq_len=1. However, in line 18 of engine.py, the output is transposed directly, changing the feature dimension to the seq_len dimension. I wonder if this handling is reasonable.