ChineseNRE

本项目使用

• python 2.7
• pytorch 0.4.0

中文实体关系抽取，对实体关系抽取不了解的可以先看这篇文章。顺便求star～

数据

中文实体关系抽取数据实在太难找了，data中是忘记在哪里找的人物关系数据集，一共11+1种关系，数据质量不太好，但也找不到其他的了。 （更新）issues7中 ybNero 同学分享了一份数据集，大家可以去issues中查看~

梅葆玥	梅兰芳	父母 坎坷经历梅葆玥之家庭合影1961年，梅兰芳先生病逝，葆玥、葆玖姐弟俩继承父亲的遗志，挑起了梅剧团的重担

数据格式为: 实体1 实体2 关系 句子。

虽然叫中文实体关系抽取，还是增加了一个英文数据集SemEval2010_task8，简单做了下数据处理，这是免费的公开数据集，其他的好像都要dollar了。。

训练

模型使用的是lstm+attention模型。特征使用词向量+位置向量。

训练前先运行data文件夹中的 data_util.py 文件，将数据处理成pkl文件供模型使用。

运行python train.py即可开始训练，可以在train.py文件中设置epoch、batch等参数，运行结束模型会储存到model文件夹中，可以在训练好的模型基础上继续训练。

可以运行python train.py pretrained使用预训练的词向量进行训练。vec.txt是一个训练好的词向量的例子，可以修改使用更加优秀的预训练词向量。

准确率

奈何实验室没有服务器，只能用自己电脑的cpu跑了一小部分数据，结果如下。

准确率	召回率	F1值
64.08%	64.59%	64.33%

参考

Attention-Based Bidirectional Long Short-Term Memory Networks for Relation Classification

更新日志

2018-10-7 第一版，不定期进行修改与优化。

2018-10-9 添加准确率、召回率、f值的计算，将model从train.py中分离。

2018-10-10 添加SemEval2010_task8数据，以及一些小修改。

2018-10-18 增加预训练的词向量，修改bug。

ChineseNRE 快速上手指南

ChineseNRE 是一个基于 LSTM+Attention 架构的中文实体关系抽取工具，支持使用预训练词向量进行模型训练。

环境准备

本项目对运行环境有特定版本要求，请确保满足以下条件：

• 操作系统：Linux / macOS / Windows (需配置相应 Python 环境)
• Python 版本：Python 2.7 (注意：项目暂不支持 Python 3)
• 深度学习框架：PyTorch 0.4.0
• 其他依赖：建议安装 numpy, pickle 等基础科学计算库

提示：由于项目较老，建议使用虚拟环境（如 virtualenv 或 conda）隔离环境，避免影响系统其他项目。

安装步骤

1. 克隆项目代码

   git clone https://github.com/buppt/ChineseNRE.git
   cd ChineseNRE
   

2. 安装依赖 由于项目年代较早，requirements.txt 可能缺失，建议手动安装指定版本的 PyTorch。

   国内用户推荐使用清华源加速安装：

   pip install -i https://pypi.tuna.tsinghua.edu.cn/simple torch==0.4.0
   pip install -i https://pypi.tuna.tsinghua.edu.cn/simple numpy
   

3. 数据预处理 在训练前，必须先运行数据处理脚本，将原始文本数据转换为模型可读取的 .pkl 格式。

   python data/data_util.py
   

基本使用

1. 开始训练

运行以下命令启动模型训练。训练完成后，模型文件将自动保存至 model 文件夹。

python train.py

注：你可以在 train.py 文件中修改 epoch、batch_size 等超参数。

2. 使用预训练词向量训练（推荐）

若希望获得更好的效果，可以使用预训练的词向量进行训练。项目自带了一个示例文件 vec.txt，你也可以替换为更优质的中文预训练向量（如 Tencent AI Lab 或 Word2Vec 中文向量）。

python train.py pretrained

3. 数据格式说明

如需添加自定义数据，请按照以下格式整理文本（实体 1、实体 2、关系类型、句子内容），并放入 data 目录后重新运行 data_util.py：

梅葆玥	梅兰芳	父母 坎坷经历梅葆玥之家庭合影 1961 年，梅兰芳先生病逝，葆玥、葆玖姐弟俩继承父亲的遗志，挑起了梅剧团的重担