Source code for WSDM 2022 paper "Few-shot Link Prediction in Dynamic Networks"
python==3.8.5
torch==1.4.0
numpy==1.21.0
scikit_learn==0.24.1
pandas==1.2.1
GPU: GeForce RTX 2080 Ti
CPU: Intel(R) Xeon(R) Silver 4210 CPU @ 2.20GHz
python code/main.py
Here, you can change hyperparameter settings in code/Config.py.
If you download datasets from urls in the paper, you need to preprocess the data by process.py or process_dblp.py.
@inproceedings{yang2022few,
title={Few-shot Link Prediction in Dynamic Networks},
author={Yang Cheng, Wang Chunchen, Lu Yuanfu, Gong Xumeng, Shi Chuan, Wang Wei and Zhang Xu},
booktitle={Proceedings of the Fifteenth ACM International Conference on Web Search and Data Mining},
pages={1245--1255},
year={2022}
}