代码:https://github.com/bojone/kg-2019
原项目是python2环境,并且模型需要自行训练。修改后的项目支持python3环境运行,并且带有训练好的模型文件,可直接运行。
@misc{
jianlin2019bdkgf,
title={A Hierarchical Relation Extraction Model with Pointer-Tagging Hybrid Structure},
author={Jianlin Su},
year={2019},
publisher={GitHub},
howpublished={\url{https://github.com/bojone/kg-2019}},
}
- JRE
- python3
词向量:https://pan.baidu.com/s/1YYE2T3f-lPyLBrJuUowAsA,密码:5p0h
Precision | Recall | F1 |
---|---|---|
0.8939 | 0.8679 | 0.8807 |
- 下载数据集,放置在
config.py
中DATASET_DIR
对应的目录下 - 下载word2vec数据集,放置在
config.py
中w2v_model_path
对应的路径位置
pip3 install requirements.txt
python3 train.py
python3 predict.py