Bert+FFN
The main requirements are:
- python 3.6
- torch 1.7.0
- tqdm
- transformers 3.5.1
Download CLUNER and put it under ./datasets/CLUENER
.
-
Get pre-trained BERT model Download bert-base-chinese and put it under
./pretrained
. -
Train
python run.py --train train
- Evaluate on the Test
python run.py --train predict
dev | test |
---|---|
76.2 | 77.5 |