spo_extract icon indicating copy to clipboard operation
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

model structure