Skip to content

Latest commit

 

History

History
29 lines (14 loc) · 711 Bytes

README.md

File metadata and controls

29 lines (14 loc) · 711 Bytes

BERT+CRF

This code is the implementation for BERT+CRF model. The implementation is based on this repo.

Requirements

  • python==3.6.9

  • torch==1.2.0

  • transformers==2.6.0

  • sklearn==0.20.2

  • seqeval

Usage

Hint: please read and delete all the comments after the backslash in each line of the .sh scripts before running them.

  1. Download LEVEN data files and put them in the data folder.
  2. Run run_train.sh for training and evaluation on the devlopment set.
  3. Run run_infer.sh to get predictions on the test set (dumped to OUTPUT_PATH/results.jsonl).

See the two scripts for more details.