FluxML / GeometricFlux.jl

Geometric Deep Learning for Flux
https://fluxml.ai/GeometricFlux.jl/stable/
MIT License
348 stars 30 forks source link

`GATConv` tests are broken by number of vertex/edge check in GraphSignals #194

Closed yuehhua closed 2 years ago

yuehhua commented 3 years ago

Mentioned in #187.

In GraphSignals v0.2 series, number of node/edge check for FeaturedGraph is introduced. Current tests will break the check due to the implementation of GATConv. GATConv has multiple head design, but the GATConv output mix the number of edges and heads together. Thus, GATConv need a new implementation to separate heads and number of edges.