Neural Collective Entity Linking Based on Recurrent Random Walk Network Learning
Models and results can be found at our IJCAI 2019 paper [Neural Collective Entity Linking Based on Recurrent Random Walk Network Learning]. It achieves the state-of-the-art result on EL task.
Details will be updated soon.
Requirement:
Python: 3.6.3
PyTorch: 0.3.1
Input format:
We transform the original data into pkl format, if you want the tranform code, please concat me.
pkl data location:
link:https://pan.baidu.com/s/17tHxyLAMqdOTozmnQsMQ6w
Fetch Code:kwvn
How to run the code?
Local:
python pre_net_xmg.py --cuda_device 0 --nohup regular --epoch 25 --weight_decay 1.28e-5 --LR 0.001 --batch 500 --filter_num 64 --filter_window 3 --local_model_loc model_loc/local/local_regular_new1 --embedding_finetune 1
Global:
python net_global_train.py --cuda_device 0 --nohup 0.5_0.1_3 --weight_decay 1.28e-5 --LR 0.0005 --local_model_loc model_loc/local/local_regular_new1.938.pkl --global_model_loc model_loc/global/global_model --random_k 3 --lamda 0.5 --flag 4:3:1 --gama 0.1 --batch 200 --epoch 25
Cite:
Please cite our IJCAI 2019 paper:
@article{xue2018,
title={Neural Collective Entity Linking Based on Recurrent Random Walk Network Learning },
author={Mengge Xue, Weiming Cai, Jinsong Su and Linfeng Song, Yubin Ge, Yubao Liu, Bin Wang},
booktitle={The Program Committee of the 28th International Joint Conference on Artificial Intelligence (IJCAI-19)},
year={2019}
}