Skip to content

LinXueyuanStdio/EchoEA

Repository files navigation

EchoEA

code for "EchoEA: Echo Information between Entities and Relations for Entity Alignment"

Environment

  • PyTorch 1.8.1 + cuda 10.2
  • apex (alternative)
  • torch-scatter, torch-sparse, torch-cluster, torch-spline-conv, torch-geometric
pip install torch-scatter -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html
pip install torch-sparse -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html
pip install torch-cluster -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html
pip install torch-spline-conv -f https://pytorch-geometric.com/whl/torch-1.8.1+cu102.html
pip install torch-geometric

How to Run

To prepare attribute similarity and value similarity:

python sim_based_on_attr_n_value.py

Complete version:

python train.py

Ablation:

  • Basic: train_basic.py
  • w/o PAN: train_basic_woPAN.py
  • w/o EN: train_basic_woEN.py
  • w/o CAN: train_basic_woCAN.py

Datasets

  • ent_ids_1: ids for entities in source KG;
  • ent_ids_2: ids for entities in target KG;
  • ref_ent_ids: entity links encoded by ids;
  • triples_1: relation triples encoded by ids in source KG;
  • triples_2: relation triples encoded by ids in target KG;

Acknowledgement

We cite the datasets and codes of these repos: GCN-Align, RREA, EMGCN, RAGA. Thanks for their great contributions!

  1. GCN-Align: paper, code
  2. RREA: paper, code
  3. RAGA: paper, code
  4. EMGCN: paper, code

About

code for "EchoEA: Echo Information between Entities and Relations for Entity Alignment"

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages