spo_extract
spo_extract copied to clipboard
基于transformers的三元组抽取
基于transformers框架的三元组抽取baseline
how to use
- 下载预训练模型,放到bert目录下,下载训练数据放到data目录下
- 安装
transformers
,pip install transformers
- 执行train.py文件
预训练模型下载地址
-
bert https://huggingface.co/bert-base-chinese/tree/main
-
roberta https://huggingface.co/hfl/chinese-roberta-wwm-ext/tree/main
训练数据下载地址
链接:https://pan.baidu.com/s/1rNfJ88OD40r26RR0Lg6Geg 提取码:a9ph