# Bert-BiLSTM-CRF-pytorch

**Repository Path**: StevenRogers/bert-bilstm-crf-pytorch

## Basic Information

- **Project Name**: Bert-BiLSTM-CRF-pytorch
- **Description**: 利用Bert-BiLSTM-CRF进行命名实体识别，深度学习框架采用pytorch.
- **Primary Language**: Unknown
- **License**: Not specified
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 9
- **Forks**: 1
- **Created**: 2021-06-16
- **Last Updated**: 2023-02-16

## Categories & Tags

**Categories**: Uncategorized

**Tags**: None

## README

# 命名实体识别

## 模型结构
* Bert + BiLSTM + CRF
## 依赖库
* transformers
* pytorch-crf
## 文件夹说明
* data/：数据集
* model/：模型代码
* train.py：训练入口
* predict：ner预测

## 训练：
1. 加载bert预训练模型
   ```shell
   cd data/
   
   git lfs install
   
   git clone https://huggingface.co/bert-base-chinese
   ```
2. 开始训练模型
    ```shell
    python main.py train --use_cuda=True --batch_size=50 --base_epoch=10
    ```