Edge weights or attention weights of trained model

benedekrozemberczki / pytorch_geometric_temporal

PyTorch Geometric Temporal: Spatiotemporal Signal Processing with Neural Machine Learning Models (CIKM 2021)

MIT License

2.61k stars 367 forks source link

Edge weights or attention weights of trained model #195

Closed Gaurangkarwande closed 1 year ago

Gaurangkarwande commented 1 year ago

Hello is there any way I can retrieve the edge weights of a StaticGraphTemporalSignal model after it has been trained. This is similar to the working of return_attention_weights() method of some of the pytorch geometric models?

benedekrozemberczki commented 1 year ago

No, but you are free to open a PR