Skip to content

sahasradude/node2vec

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

48 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

To test the link prediction task on the test set of the last 25% of links formed in the graph, run:

python3 test.py

To run it on the larger reddit dataset, in test.py, change the dataset path and the training set path to: "../datasets/redditdataset_75.txt", "../datasets/redditdataset_test.txt"

in the run function call.

The output of the function gives two lines; the number of links successfully predicted, and the Mean Average Precision @ 10 for the ranking of links returned for each node

About

Implementation of the node2vec algorithm.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Contributors

Languages

  • Python 100.0%