/PyTorch_BERT_Pipeline_IE

复习论文《A Frustratingly Easy Approach for Joint Entity and Relation Extraction》

Primary LanguagePythonMIT LicenseMIT

PyTorch_BERT_Pipeline_IE

概述

1、复习论文《A Frustratingly Easy Approach for Joint Entity and Relation Extraction》
2、模型的具体思路描述可以见知乎
3、训练数据来自于百度的2021LIC语言与计算智能竞赛,没有上传数据,可以自行到官网下载哦。

环境要求

pytorch >=1.6.0
transformers>=3.4.0

运行步骤

1、进入entity_extract项目或relationship_classifiction项
2、去huggingface官网下载BERT预训练权重,然后并放在./pretrained_model/文件夹下
3、在./utils/arguments_parse.py中修改BERT预训练模型的路径
4、运行train.py进行训练
5、运行predict.py进行预测