How to run BERT-CRF baseline?

xuuuluuu / SynLSTM-for-NER

Code and models for the paper titled "Better Feature Integration for Named Entity Recognition", NAACL 2021.

30 stars 2 forks source link

How to run BERT-CRF baseline? #3

Closed Elanmarkowitz closed 2 years ago

Elanmarkowitz commented 2 years ago

Hi, how do I run the BERT-CRF baseline? I only see ["none", "dggcn", "dglstm"] as options in the parser.

Thanks!

xuuuluuu commented 2 years ago

Hi, thanks for following our work. The current setup can support BERT-CRF but requires some simple modifications. I will arrange my time to do the modification. Meanwhile, I also used the repo below to check the BERT-CRF performance. https://github.com/allanj/pytorch_neural_crf