transformers-ner
transformers-ner copied to clipboard
Pytorch-Named-Entity-Recognition-with-transformers
transformers-ner
Pytorch-Named-Entity-Recognition-with-transformers
Support chinese ner and english ner datasets
dataset list
- cluener(chinese)
- conll2003(english)
model_list
- (bert, albert, electra, roberta, distilbert, camembert, xlm, xlmroberta) + softmax
- (bert, albert, electra, roberta, distilbert, camembert, xlm, xlmroberta) + crf
- (bert, albert, electra, roberta, distilbert, camembert, xlm, xlmroberta) + span
requirement
- Pytorch=1.1.0+
- transformers=2.8.0
run the code
-
sh scripts/run_xxx_ner.sh