Add GINConv support to HeteroGraphConv

CarloLucibello / GraphNeuralNetworks.jl

Graph Neural Networks in Julia

MIT License

220 stars 46 forks source link

Closed rbSparky closed 7 months ago

rbSparky commented 7 months ago

Covers Issue #311

This operation needs to be done differently for hetero graphs

    if g isa GNNHeteroGraph
        #
    else
        l.nn((1 + ofeltype(x, l.ϵ)) * x + m)
    end

[EDIT: done]

rbSparky commented 7 months ago

ready for merge?

rbSparky commented 7 months ago

Changes done