# RHKH
## Dependencies  

* Compatible with PyTorch 1.7.1 and Python 3.7.  

```
msgpack_numpy==0.4.7.1
msgpack_python==0.5.6
natsort==8.1.0
numpy==1.19.2
ordered_set==4.1.0
torch==1.7.1
torch_scatter==2.0.5
tqdm==4.59.0
```

## Dataset

* Datasets are included in the `data` directory.  

## Training Model

* Commands for reproducing the reported results on link prediction:
  * Train and Test on JF17K:
    * ``python run.py -score_func conve  -opn corr  -gpu 0 -epoch 500 -norm 2 -batch 256 -margin 0 -neg 400 -r2fc 0  -dist_type 0 -multi_r 1 -position 1 -shift 1  -gcn_drop 0.0 -id_drop 0.2 -feat_drop 0.2 -data data/JF17K`` 

  * Train and Test on FB-AUTO:
    * ``python run.py -score_func conve  -opn corr  -gpu 0 -epoch 500 -norm 2 -batch 256 -margin 0 -neg 400 -r2fc 0  -dist_type 0 -multi_r 1 -position 1 -shift 1  -gcn_drop 0.0 -id_drop 0.2 -data data/FB-AUTO`` 

  * Train and Test on M-FB15K:
    * ``python run.py -score_func conve  -opn corr  -gpu 0 -epoch 500 -norm 2 -batch 256 -margin 0 -neg 400 -r2fc 0  -dist_type 0 -multi_r 1 -position 1 -shift 1  -gcn_drop 0.0 -id_drop 0.2 -data data/M-FB15K`` 

