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.
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
Demo datasets are available in the folder-input.
A demo file is given to show the execution of model and other tools.
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},
}