/ChineseNER_BERT

🚀 基于PyTorch的BERT实体识别框架

Primary LanguagePython

PyTorch BERT中文实体识别

预训练模型下载路径

https://huggingface.co/bert-base-chinese/tree/main

下载config.json, pytorch_model.bin, vocab.txt, 存放在pretrained/bert-base-chinese/文件夹中

pretrained
│  
└─bert-base-chinese
        config.json
        pytorch_model.bin
        vocab.txt

环境

torch==1.10.1+cu113
transformers==4.15.0
argparse==1.4.0
numpy==1.22.3
seqeval==1.2.2

使用方法

  1. 安装环境
pip install requirements.txt
  1. 运行代码
python main.py

欢迎star