search

PetarV- / DGI

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

Codes on reddit dataset #2

Closed Tiiiger closed 5 years ago

Tiiiger commented 5 years ago

Great work! I really enjoy reading the paper.

However, will the codes to replicate the reported performance on reddit be released? If so, is there a planned schedule?

Thank you!

PetarV- commented 5 years ago

Thank you for the kind words!

Currently, there are no such plans --- but I can confirm that we used GraphSAGE's TensorFlow repository:

https://github.com/williamleif/GraphSAGE

as a starting point.

This is due to the fact that the PyTorch version of GraphSAGE did not seem to work with Reddit.