Skip to content

Latest commit

 

History

History

BERT+CRF

Folders and files

NameName
Last commit message
Last commit date

parent directory

..
performance
performance
 
 
README.md
README.md
 
 
bert_crf.py
bert_crf.py
 
 
crf.py
crf.py
 
 
run_infer.sh
run_infer.sh
 
 
run_leven.py
run_leven.py
 
 
run_train.sh
run_train.sh
 
 
utils_leven.py
utils_leven.py
 
 

README.md

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.