dgl-ke使用方法和参数

https://dglke.dgl.ai/doc/emb_sim.html

dglke_train --model_name TransR --data_path data/own_kg --data_files train.txt test.txt valid.txt --format raw_udd_hrt --dataset own_kg --batch_size 1000 --log_interval 100 --neg_sample_size 200 --regularization_coef=1e-9 --hidden_dim 400 --gamma 19.9 --lr 0.25 --batch_size_eval 16 --test -adv --gpu 1 --max_step 20000

dglke_predict --model_path /root/lv_my_task/ckpts/TransE_l1_own_kg_0 --format 'h_r_*' --data_files head.list rel.list --score_func logsigmoid --topK 5 --raw_data --entity_mfile /root/lv_my_task/data/own_kg/entities.tsv --rel_mfile /root/lv_my_task/data/own_kg/relations.tsv

dglke_emb_sim --emb_file /root/lv_my_task/ckpts/TransE_l1_own_kg_0/own_kg_TransE_l1_entity.npy --format 'l_*' --data_files head.list  --topK 50 --raw_data --mfile /root/lv_my_task/data/own_kg/entities.tsv