hazdzz / STGCN

The PyTorch implementation of STGCN.
GNU Lesser General Public License v2.1
484 stars 106 forks source link

Is there a way to use other feature data together? #6

Closed jeongwhanchoi closed 3 years ago

jeongwhanchoi commented 3 years ago

Is there a way to use feature data other than velocity data together? I'd like to use other location features (e.g. landmark, the num of houses, etc.).

The STGCN models I know use only one feature for the speed of each node, but I would like to ask if there is any idea of using other information of the node as well.

Thank you.

hazdzz commented 3 years ago

Sorry, STGCN only accepts one-dimensional feature data. That's why I'm studying other spatio-temporal graph convolution neural network models.