FreeRotate/ChineseNER

⚡ 基于PyTorch的多种模型实体识别框架

Pytorch的中文命名实体识别

包含的模型如下

LSTM BiLSTM

GRU BiGRU

环境

torch==1.10.1+cu113
argparse==1.4.0
numpy==1.22.3
seqeval==1.2.2
pandas==1.1.1

使用方法

1. 安装环境

pip install requirements.txt

2. 运行代码

python main.py