BERT_ChineseWordSegment

所属分类:人工智能/神经网络/深度学习
开发工具:Python
文件大小:2952KB
下载次数:0
上传日期:2019-05-28 07:47:31
上 传 者sh-1993
说明:  BERT_ChineseWordSegment,一个基于BERT的中文分词模型,F1得分97%
(BERT_ChineseWordSegment,A Chinese word segment model based on BERT, F1-Score 97%)

文件列表:
LICENSE (11357, 2019-05-28)
PEOPLEdata (0, 2019-05-28)
PEOPLEdata\test.txt (1461285, 2019-05-28)
PEOPLEdata\train.txt (5835809, 2019-05-28)
evaluation.py (3246, 2019-05-28)
run_cut.py (24873, 2019-05-28)

# BERT_ChineseWordSegment Try to implement a Chinese word segment work based on Google BERT! The corpus is extracted from The People's Daily (Chinese: 人民日报, Renmin Ribao).
First git clone https://github.com/google-research/bert.git Second put the three scripts: modeling.py、optimization.py、tokenization.py into this project, structure is as follows: BERT_ChinesewordSegment |____ PEOPLEdata |____ output |____ modeling.py |____ optimization.py |____ tokenization.py |____ run_cut.py |____ evaluation.py Third download the Chinese pre-trained bert model [BERT-Base, Chinese](https://storage.googleapis.com/bert_models/2018_11_03/chinese_L-12_H-768_A-12.zip) And then set pre-trained model path and data path environment: $BERT_CHINESE_DIR、$PEOPLEcut ## run ``` python3 run_cut.py --task_name="people" --do_train=True --do_predict=True --data_dir=$PEOPLEcut --vocab_file=$BERT_CHINESE_DIR/vocab.txt --bert_config_file=$BERT_CHINESE_DIR/bert_config.json --init_checkpoint=$BERT_CHINESE_DIR/bert_model.ckpt --max_seq_length=128 --train_batch_size=32 --learning_rate=2e-5 --num_train_epochs=3.0 --output_dir=./output/result_cut/ ``` It will take about 28 minutes with 3 epochs on a GPU. This will produce an evaluate output like this: ``` INFO:tensorflow:***** Eval results ***** INFO:tensorflow: count = 9925 INFO:tensorflow: precision_avg = 0.9794 INFO:tensorflow: recall_avg = 0.9780 INFO:tensorflow: f1_avg = 0.9783 INFO:tensorflow: error_avg = 0.0213 ``` And the word segmentation results will be seen in ./output/result_cut/seg_result.txt If you want learn more details, see the code analysis(in Chinese)[简书:BERT系列(五)——中文分词实践...](https://www.jianshu.com/p/be0a951445f4)

近期下载者

相关文件


收藏者