working-yuhao / DEAL

IJCAI2020
MIT License
19 stars 7 forks source link

DEAL

This is the implementation of "Inductive Link Prediction for Nodes Having Only Attribute Information " published at IJCAI 2020.

Environment

pytorch 1.4.0

python 3.7.4

pytorch-geometric 1.4.3

cuda 10.1

Train DEAL

python train.py

Datasets

More datasets can be found at https://pytorch-geometric.readthedocs.io/en/latest/modules/datasets.html.

Cite

Please cite our IJCAI 2020 paper:

@inproceedings{ijcai2020-168,
  title     = {Inductive Link Prediction for Nodes Having Only Attribute Information},
  author    = {Hao, Yu and Cao, Xin and Fang, Yixiang and Xie, Xike and Wang, Sibo},
  booktitle = {Proceedings of the Twenty-Ninth International Joint Conference on
               Artificial Intelligence, {IJCAI-20}},
  publisher = {International Joint Conferences on Artificial Intelligence Organization},             
  editor    = {Christian Bessiere}, 
  pages     = {1209--1215},
  year      = {2020},
  month     = {7},
  note      = {Main track}
  doi       = {10.24963/ijcai.2020/168},
  url       = {https://doi.org/10.24963/ijcai.2020/168},
}