← all repositories

lonePatient/BERT-NER-Pytorch

Chinese NER system using BERT models with Softmax, CRF, and Span-based decoding layers.

2.2k stars Python Domain AppsLanguage Models
BERT-NER-Pytorch
Velocity · 7d
+0.8
★ / day
Trend
steady
star history

This repository implements named entity recognition for Chinese text using BERT and ALBERT pre-trained models. It provides three decoding approaches: Softmax classification, conditional random field (CRF) layers, and span-based extraction. The project supports adversarial training, focal loss, and label smoothing for improved performance. It evaluates on CLUENER and CNER datasets with benchmark F1 scores around 0.81 for BERT+Span.

heatdrop uses Google Analytics to see which pages get read — nothing else. Your call. How we handle data.