gammagl.models.DGIModel

class DGIModel(in_feat, hid_feat, act)

Deep Graph Infomax in DGL

References

Papers: https://arxiv.org/abs/1809.10341 Author’s code: https://github.com/PetarV-/DGI

forward(feat, edge_index, edge_weight, num_nodes)