NeuLP: An End-to-end Deep-learning Model for Link Prediction

Zhiqiang Zhong, Yang Zhang, Jun Pang. NeuLP: An End-to-end Deep-learning Model for Link Prediction. 21th International Conference on Web Information Systems Engineering (WISE’20).

Implementation of NeuLP with Pytorch.

[Paper]

Required packages

The code has been tested running under Python 3.7.1. with the following packages installed (along with their dependencies):

• numpy == 1.18.1
• pandas == 1.0.3
• scikit-learn == 0.22.2
• networkx == 2.4
• pytorch == 1.4.0
• torch_geometric == 1.4.2

Data requirement

Demo datasets are available in the folder-input.

Code execution

A demo file is given to show the execution of model and other tools.

Citation

Please cite our paper if you make use of this code in your own work:

@article{ZhongZP2020,
  title={NeuLP: An End-to-end Deep-learning Model for Link Prediction},
  author={Zhiqiang Zhong, Yang Zhang, Jun Pang},
  booktitle={Web Information Systems Engineering - {WISE} 2020 - 21st International Conference},
  pages={96--108},
  year={2020},
}