phanein / deepwalk

DeepWalk - Deep Learning for Graphs
http://www.perozzi.net/projects/deepwalk/
Other
2.69k stars 825 forks source link

How to predict new links in graph using skipgram output? #70

Closed Anjs04 closed 6 years ago

Anjs04 commented 6 years ago

I found this guide using node2vec: https://github.com/lucashu1/link-prediction/blob/master/node2vec.ipynb

It shows how to create edge embeddings and apply logistic regression in steps 4 and 5. This could be applied to the embeddings outputted by the deepwalk algorithm.