Nealcly / templateNER

Source code for template-based NER
208 stars 39 forks source link
few-shot-learning named-entity-recognition prompt sequence-labeling template

Template-Based NER

Source Code For Template-Based Named Entity Recognition Using BART

Training

Training train.py

Inference inference.py

Corpus

ATIS (https://github.com/yvchen/JointSLU/tree/master/data)

MIT Restaurant Corpus (https://groups.csail.mit.edu/sls/downloads/)

MIT Movie Corpus (https://groups.csail.mit.edu/sls/downloads/)

Contact

If you have any questions, please feel free to contact Leyang Cui (cuileyang@westlake.edu.cn).

Citation

@inproceedings{cui-etal-2021-template,
    title = "Template-Based Named Entity Recognition Using {BART}",
    author = "Cui, Leyang  and
      Wu, Yu  and
      Liu, Jian  and
      Yang, Sen  and
      Zhang, Yue",
    booktitle = "Findings of the Association for Computational Linguistics: ACL-IJCNLP 2021",
    month = aug,
    year = "2021",
    address = "Online",
    publisher = "Association for Computational Linguistics",
    url = "https://aclanthology.org/2021.findings-acl.161",
    doi = "10.18653/v1/2021.findings-acl.161",
    pages = "1835--1845",
}