- Tensorflow-gpu==1.12.0
- Download BERT-Base中文模型 or Chinese-BERT-wwm
- Format (seperate by '\t' rather than ',') dataset like:
train.txt: 1 ChineseTextChineseText 0 ChineseTextChineseTextChineseText 0 ChineseText 1 ChineseTextChineseTextChineseTextChineseText eval.txt: 1 ChineseTextChineseText 0 ChineseTextChineseTextChineseText 0 ChineseText 1 ChineseTextChineseTextChineseTextChineseText test.txt 0 ChineseTextChineseTextChineseText 0 ChineseTextChineseTextChineseText 0 ChineseTextChineseTextChineseText 0 ChineseTextChineseTextChineseText
- Set runtime paraments in
arguments.py
. - training: 'do_predict' set to False, 'do_train' and 'do_eval' set to True in
arguments.py
, and runtrain_eval.py
. - predict: 'do_predict' set to True, 'do_train' and 'do_eval' set to False in
arguments.py
, and runtrain_eval.py
. - Find the results file in
./output/test_results.tsv
.