benedekrozemberczki / pytorch_geometric_temporal

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)
MIT License
2.63k stars 370 forks source link

HeteroGCLSTM for edge regression #241

Open MarcoLomele opened 1 year ago

MarcoLomele commented 1 year ago

Hi everyone,

How can I pass edge features to a HeteroGCLSTM layer? Looking at the documentation it seems that only node features can be passed.

I’m researching the use of graph neural networks for predicting economical interactions between countries. My goal is to predict edge weights (called labels in PyG's documentation), given nodes' features and edges' attributes. So far, I repreesnt temporal relations only through the input features. As such, my model considers monthly snapshots individually during the training and testing phase.

Huge thanks to @benedekrozemberczki for making this library available to everyone!