Skip to content

An open-source highly heterogeneous entity alignment (HHEA) toolkit.

Notifications You must be signed in to change notification settings

jxh4945777/OpenHHEA

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

1 Commit
 
 
 
 
 
 

Repository files navigation

OpenHHEA : Toolkit for Entity Alignment

Code of OpenHHEA

Requirements

python >= 3.9
scipy
numpy
networkx
pillow
gensim
openai
transformers
torch

Main Interface

The main interface provided by OpenHHEA is included as follows:

  • OpenHHEA.get_model_config: get the predifined configurations of model and data processing.
  • OpenHHEA.KGDataLoader: the data loader for loading knowledge graph data.
  • OpenHHEA.KGDataProcessor: the processor for process the knowledge graph data to get feature embeddings of name information, structure information, image information and so on.
  • OpenHHEA.HHEATrainer: the trainer for training embedding-based entity alignment model. Use train function to start training.
  • OpenHHEA.ReasoningXXX: the encapsulated entity alignment reasoning process. Use run function to start entity alignment reasoning.
  • OpenHHEA.init_model_pipeline: get the predefined entity alignment pipeline, including the training and reasoning. Use run to start the whole entity alignment pipeline.

Usage

For the detailed usage of OpenHHEA, including the methods for using customized reasoning workflow and using redefined reasoning workflow, please see example_code.py

About

An open-source highly heterogeneous entity alignment (HHEA) toolkit.

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

 
 
 

Languages