Need help with ChinsesNER-pytorch?
Click the “chat” button below for chat support from the developer who created it, or find similar developers for support.

About the developer

yanwii
184 Stars 57 Forks 14 Commits 4 Opened issues

Description

基于BI-LSTM+CRF的中文命名实体识别 Pytorch

Services available

!
?

Need anything else?

Contributors list

# 103,214
Python
bert-bi...
Tensorf...
beam-se...
1 commit

ChinsesNER-pytorch

train

setp 1: edit models/config.yml

embedding_size: 100
hidden_size: 128
model_path: models/
batch_size: 20
dropout: 0.5
tags:
  - ORG
  - PER

step 2: train

python3 main.py train
or
cn = ChineseNER("train")
cn.train()

... epoch [4] |██████ | 154/591 loss 0.46 evaluation ORG recall 1.00 precision 1.00 f1 1.00


epoch [4] |██████ | 155/591 loss 1.47 evaluation ORG recall 0.92 precision 0.92 f1 0.92


epoch [4] |██████ | 156/591 loss 0.46 evaluation ORG recall 0.94 precision 1.00 f1 0.97

predict

python3 main.py predict
or 
cn = ChineseNER("predict")
cn.predict()

请输入文本: 海利装饰材料有限公司 [{'start': 0, 'stop': 10, 'word': '海利装饰材料有限公司', 'type': 'ORG'}]

REFERENCES

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.