We use entity alignment benchmark datasets OpenEA which can be downloaded from OpenEA. You need to put the prepared data into /data folder.
- Python 3
- PyTorch
- networkx==2.5.1
- Scipy
- Numpy
- Pandas
- Scikit-learn
- Faiss
You can automatically download corresponding dependencies by following scripts:
pip install -r .\requirements.txt
Note: The settings of hyper-parameters are given in /args folder.
To run MixTEA, please use the following scripts (ps: --task is an argument):
python run.py --task en_fr_15k
python run.py --task en_de_15k
python run.py --task d_w_15k
python run.py --task d_y_15k
To run 5-fold cross-validation, please use the following script:
python run_fold.py --task en_fr_15k
We also provide jupyter notebook version in MixTEA.ipynb.
If you have any difficulty or question in running code and reproducing experimental results, please email to xiefeng@nudt.edu.cn.
We refer to the codes of these repos: GCN-Align, OpenEA, MuGNN, MeanTeacher. Thanks for their great contributions!