relation-extract-R-Bert

【文件属性】：

文件名称：relation-extract-R-Bert

文件大小：172KB

文件格式：ZIP

更新时间：2024-05-01 05:23:53

Python

R-BERT R-BERT是用分类的思想实现关系抽取。与基于BERT模型的分类模型有一些区别： 加入实体的标志符号，在BERT提取特征后，又提取了两个实体的词向量作为特征； 思想比较简单，模型的效果很好。 数据集 ccks2019关系抽取的数据集。将数据集处理为下面的格式： 出生日期 \t 汪晋贤，1996年1月2日出生于湖北省武汉市，身高175cm，是一名中场球员，现效力于大连一方足球俱乐部 代码结构 |__main.py |__data_loader.py |__model.py |__trainer.py |__utils.py |__export_onnx.py |__predict.py 模型结果

【文件预览】：
relation-extract-R-Bert-master
----ReadMe.md(712B)
----R-BERT()
--------prepare_chinese_data_for_r_bert.py(2KB)
--------main.py(5KB)
--------utils.py(2KB)
--------data()
--------data_loader.py(11KB)
--------ReadMe.md(705B)
--------predict.py(7KB)
--------torch2tfmodel.py(400B)
--------LICENSE(11KB)
--------trainer_new.py(10KB)
--------.idea()
--------trainer.py(10KB)
--------requirements.txt(32B)
--------model.py(4KB)
--------export_onnx.py(5KB)
--------rbert_config.py(6KB)