Skip to content

Latest commit

 

History

History
37 lines (29 loc) · 1.56 KB

README.md

File metadata and controls

37 lines (29 loc) · 1.56 KB
Raw

HousE: Knowledge Graph Embedding with Householder Parameterization

This is the code of the paper HousE: Knowledge Graph Embedding with Householder Parameterization for ICML 2022.

A more powerful and general framework for knowledge graph embedding.

Requirements

  • pytorch == 1.8.0
  • numpy == 1.19.2
  • scikit-learn == 0.23.2

Data

  • entities.dict: a dictionary map entities to unique ids
  • relations.dict: a dictionary map relations to unique ids
  • train.txt: the KGE model is trained to fit this data set
  • valid.txt: create a blank file if no validation data is available
  • test.txt: the KGE model is evaluated on this data set

Models

  • HousE-r: relational Householder rotations
  • HousE: relational Householder projections + relational Householder rotations
  • HousE-r+: relational Householder rotations + translations
  • HousE+: relational Householder projections + relational Householder rotations + translations

Usage

All training commands are listed in best_config.sh. For example, you can run the following commands to train HousE on WN18RR and FB15k-237 datasets.

# WN18RR
bash run.sh HousE wn18rr 0 0 0 1000 200 800 8 1 0.5 6.0 1.14940435933987 0.000575323908649059 60000 20000 8 0.0960737047401994

# FB15k-237
bash run.sh HousE FB15k-237 0 0 0 500 500 600 20 6 0.6 5.0 2.00378388680359 0.000794267891285676 100000 10000 16 0.00336727231946076

Acknowledgement

We refer to the code of RotatE. Thanks for their contributions.