Bert-BiLSTM-CRF-pytorch-master (1)
所属分类:界面编程
开发工具:Python
文件大小:4042KB
下载次数:0
上传日期:2021-03-19 16:09:01
上 传 者:
here.senthil
说明: named entity retrival python code
文件列表:
LICENSE (1068, 2019-12-27)
config.py (372, 2019-12-27)
crf.py (6818, 2019-12-27)
crf_predict.py (3072, 2019-12-27)
data (0, 2019-12-27)
data\example.dev (702455, 2019-12-27)
data\example.test (1405788, 2019-12-27)
data\example.train (6277429, 2019-12-27)
data_process.ipynb (175852, 2019-12-27)
main.py (6163, 2019-12-27)
model.py (1509, 2019-12-27)
pred.ipynb (26712, 2019-12-27)
processed (0, 2019-12-27)
processed\processed_dev_bio.txt (310401, 2019-12-27)
processed\processed_test.txt (843426, 2019-12-27)
processed\processed_test_bio.txt (2656793, 2019-12-27)
processed\processed_training.txt (2656793, 2019-12-27)
processed\processed_training_bio.txt (2346392, 2019-12-27)
utils.py (3923, 2019-12-27)
# Bert-BiLSTM-CRF-pytorch
bert-bilstm-crf implemented in pytorch for named entity recognition.
```
python == 3.6
pytorch == 1.3
transformer == 2.1.1
```
### Data
* 首先将数据处理成`BIO`格式,processed文件夹下存放的是医疗命名实体识别的数据,代码可参考`data_process.ipynb`
* 下载中文BERT预训练模型,来自`pytorch-pretrained-bert`
### 模型训练
```
python main.py -- n_epochs 100 --finetuning --top_rnns
```
### 模型预测
```
python crf_predict.py
```
近期下载者:
相关文件:
收藏者: