【nlp pytorch】基于标注信息从句子中提取命名实体内容
基于标注信息从句子中提取实体内容
- 1 需求
- 2 代码实现
- 3 代码封装
1 需求
给定一个句子和已经通过模型训练标注好的信息，从而提取出句子中的实体内容，如下
输入：
（1）句子信息
每个糖尿病患者,无论是病情轻重,不论是注射胰岛素,还是口服降糖药,都必须合理地控制饮食。
（2）标注信息
['O', 'O', 'B-Disease', 'I-Disease', 'I-Disease', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'O', 'B-Drug', 'I-Drug', 'I-Drug', 'O', 'O', 'O', 'B-Drug', 'I-Drug', 'I-Drug', 'I-Drug', 'I-Drug', 'O', 'O', 'O', 'O', 'O', 'O'