extended GatedGraphConv and NNConv to use AbstactGNNGraph#375
extended GatedGraphConv and NNConv to use AbstactGNNGraph#375Ghaithq wants to merge 1 commit intoJuliaGraphs:masterfrom
Conversation
|
Hi @CarloLucibello. Is this correct or did I misunderstand something? |
|
Thanks for this contribution! It would be nice to have some tests |
|
Should I modify the already made tests for both NNConv and GatedGraphConv to use AbstactGNNGraph. or should I implement new tests. And I was planning on resolving the whole issue should I continue in the same manner? |
| @non_differentiable fill!(x...) | ||
|
|
||
| function (l::GatedGraphConv)(g::GNNGraph, H::AbstractMatrix{S}) where {S <: Real} | ||
| function (l::GatedGraphConv)(g::AbstractGNNGraph, H::AbstractMatrix{S}) where {S <: Real} |
There was a problem hiding this comment.
just this change is not enough to make this layer compatible with heterograph
There was a problem hiding this comment.
Ok, could you tell me what I should add?
There was a problem hiding this comment.
and is NNConv ok or should I also modify it?
you can add tests here |
This is an attempt in solving #311 partially by extending both NNConv and GatedGraphConv.