search

liuxiyang641 / RAGAT

RAGAT: Relation Aware Graph Attention Network for Knowledge Graph Completion
56 stars 16 forks source link

Hyperparameter on WN18RR dataset #1

Closed junkangwu closed 3 years ago

junkangwu commented 3 years ago

May I ask you about the hyperparameters on WN18RR? Thanks a lot!

liuxiyang641 commented 3 years ago

You can try the following command:

python run.py -epoch 1500 -name test_wn -model ragat -score_func interacte -opn cross -gpu 0 -data WN18RR -gcn_drop 0.4 -ifeat_drop 0.2 -ihid_drop 0.3 -batch 256 -iker_sz 11 -iperm 4 -attention True -head_num 1