Skip to content

Pytorch-Named-Entity-Recognition-with-transformers

Notifications You must be signed in to change notification settings

huangjiaheng/transformers-ner

 
 

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

17 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

transformers-ner

Pytorch-Named-Entity-Recognition-with-transformers

Support chinese ner and english ner datasets

dataset list

  1. cluener(chinese)
  2. conll2003(english)

model_list

  1. (bert, albert, electra, roberta, distilbert, camembert, xlm, xlmroberta) + softmax
  2. (bert, albert, electra, roberta, distilbert, camembert, xlm, xlmroberta) + crf
  3. (bert, albert, electra, roberta, distilbert, camembert, xlm, xlmroberta) + span

requirement

  1. Pytorch=1.1.0+
  2. transformers=2.8.0

run the code

  1. sh scripts/run_xxx_ner.sh

About

Pytorch-Named-Entity-Recognition-with-transformers

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages

  • Python 98.2%
  • Shell 1.8%