PyTorch implementation of ComplexGCN:
CUDA_VISIBLE_DEVICES=1 python complexgcn.py -model_name complexgcn -dataset FB15k-237
CUDA_VISIBLE_DEVICES=1 python complexgcn.py -model_name complexgcn -dataset WN18RR -lr 0.0003 -dim 400 -drop1 0.3 -drop2 0.3 -drop3 0.3
For ablated model on FB15k-237, set -model_name ablated
Unzip data.zip
#Required packages:
Python 3.7.6
numpy 1.19.2
pytorch 1.8.1
Acknowledgment: The code is based on https://github.com/tkipf/pygcn, https://github.com/ibalazevic/TuckER, https://github.com/TimDettmers/ConvE