Paper: "End-to-end Structure-Aware Convolutional Networks for Knowledge Base Completion"
Published in the Thirty-Third AAAI Conference on Artificial Intelligence (AAAI-19).
--- PyTorch Version ---
The end-to-end Structure-Aware Convolutional Network (SACN) model takes the benefit of GCN and ConvE together for knowledge base completion. SACN consists of an encoder of a weighted graph convolutional network (WGCN), and a decoder of a convolutional network called Conv-TransE. WGCN utilizes knowledge graph node structure, node attributes and edge relation types. The decoder Conv-TransE enables the state-of-the-art ConvE to be translational between entities and relations while keeps the same link prediction performance as ConvE.
This repo supports Linux and Python installation via Anaconda.
-
Install PyTorch 1.0 using official website or Anaconda.
-
Install the requirements:
pip install -r requirements.txt -
Download the default English model used by spaCy, which is installed in the previous step
python -m spacy download en.
Run the preprocessing script for FB15k-237, WN18RR, FB15k-237-attr and kinship: sh preprocess.sh.
To run a model, you first need to preprocess the data. This can be done by specifying the process parameter.
For ConvTransE model, you can run it using:
CUDA_VISIBLE_DEVICES=0 python main.py model ConvTransE init_emb_size 100 dropout_rate 0.4 channels 50 lr 0.001 kernel_size 3 dataset FB15k-237 process True
For SACN model, you can run it using:
CUDA_VISIBLE_DEVICES=0 python main.py model SACN dataset FB15k-237 process True
You can modify the hyper-parameters from "src.spodernet.spodernet.utils.global_config.py" or specify the hyper-parameters in the command. For different datasets, you need to tune the parameters.
For this test version, if you find any problems, please feel free and email me. We will keep updating the code.
Code is inspired by ConvE.