/NeuLP

[WISE20] NeuLP: An End-to-end Deep-learning Model for Link Prediction (https://dl.acm.org/doi/abs/10.1007/978-3-030-62005-9_8)

Primary LanguageJupyter Notebook

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},
}