Transfer-and-Fusion: Integrated Link Prediction across Knowledge Graphs

Quick Start

Install PyTorch following the instructions on the PyTorch. Our code is written in Python3.

conda install pytorch==1.11.0 torchvision==0.12.0 torchaudio==0.11.0 cudatoolkit=11.3 -c pytorch
conda install pyg -c pyg

python main.py -dataset <dataset_name> -gpu <device_id>

In our experiments, the datasets used are: DBP-FB or WIKI-YAGO.

Name		Name	Last commit message	Last commit date
Latest commit History 2 Commits
src		src
LICENSE		LICENSE
README.md		README.md
evaluation.py		evaluation.py
main.py		main.py