/PHGCN

Pair-Wise or High-Order? A Self-Adaptive Graph Framework for Knowledge Graph Embedding

This repository provides the official PyTorch implementation of the research paper Pair-Wise or High-Order? A Self-Adaptive Graph Framework for Knowledge Graph Embedding. If the paper is officially published, we will release all the experimental codes. It will update soon. Thanks for your attention.

Requirements

  • python >= 3.6
  • pytorch == 1.8.1
  • torch-geometric == 2.0.3
  • torch-scatter == 2.0.6
  • torch-sparse == 0.6.12
  • torchsummary == 1.5.1
  • tqdm == 4.63.0
  • simcse == 0.4
  • scikit-learn == 0.21.1
  • numpy == 1.19.5

How to run

PHGCN:

python run.py -use_scnn -edge_sample_scale 8000 -lr 0.001 -data WN18RR -score_func conve