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
进行预测