nnzhan / Graph-WaveNet

graph wavenet
MIT License
628 stars 201 forks source link

How to modify the feature_dimension of the output from `gwnet` to be 2, such us (64,12,207,2) #40

Open HQ-LV opened 8 months ago

HQ-LV commented 8 months ago

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.