search

PetarV- / DGI

Deep Graph Infomax (https://arxiv.org/abs/1809.10341)
MIT License
630 stars 135 forks source link

Script for inductive training #12

Open kavehhassani opened 4 years ago

kavehhassani commented 4 years ago

Hi,

Can you please share your code for training and evaluating PPI dataset as well?

Thanks

Haicang commented 4 years ago

Also, could you please share your code for large dataset like reddit? As in the paper, sampling is essential to avoid huge GPU memory usage, but it is not in this repo.

mgparkdrama commented 3 years ago

Also, could you please share your code for large dataset like reddit? As in the paper, sampling is essential to avoid huge GPU memory usage, but it is not in this repo.

You can use GraphSAGE in DGL for mini-batch training.