search

CarloLucibello / GraphNeuralNetworks.jl

Graph Neural Networks in Julia
https://carlolucibello.github.io/GraphNeuralNetworks.jl/dev/
MIT License
210 stars 47 forks source link

Added `TAGConv` layer #430

Closed rbSparky closed 2 weeks ago

rbSparky commented 3 months ago

Implements Topology Adaptive Graph Convolutional Networks

Computes:

$$ H^{K} = \sum{k=0}^K (D^{-1/2} A D^{-1/2})^{k} X \Theta{k} $$

let me know if there are changes required in the method of computation

rbSparky commented 1 week ago

If further changes to this are needed I will open a PR to fix them. I will test this implementation and compare it to DGL once