# Bert-BiLSTM-CRF-pytorch-master

**Repository Path**: jimhamburger/keywords_-extraction

## Basic Information

- **Project Name**: Bert-BiLSTM-CRF-pytorch-master
- **Description**: No description available
- **Primary Language**: Python
- **License**: Apache-2.0
- **Default Branch**: master
- **Homepage**: None
- **GVP Project**: No

## Statistics

- **Stars**: 0
- **Forks**: 0
- **Created**: 2021-11-24
- **Last Updated**: 2021-12-07

## Categories & Tags

**Categories**: Uncategorized

**Tags**: 序列标注, 关键词提取, bert, LSTM, CRF

## README

# Bert-BiLSTM-CRF-pytorch
bert-bilstm-crf implemented in pytorch for named entity recognition.

```
python == 3.6
pytorch == 0.4.1
pytorch_pretrained_bert == 0.6.1
```

### Data
* 首先将数据处理成`BIO`格式，processed文件夹下存放的是医疗命名实体识别的数据，代码可参考`data_process.ipynb`
* 下载中文BERT预训练模型,来自`pytorch-pretrained-bert`

### 模型训练
```
python main.py -- n_epochs 100 --finetuning --top_rnns
```


### 模型预测
```
python crf_predict.py
```