Skip to content

Latest commit

 

History

History
19 lines (12 loc) · 636 Bytes

README.md

File metadata and controls

19 lines (12 loc) · 636 Bytes

利用bert预训练的中文语言模型进行文本分类

训练脚本 train.sh

批量测试脚本 predict.sh

利用模型进行单条语句测试 intent.py 注:修改single_predict.py中 get_test_examples,get_labels方法 max_seq_length,需run_classifier.py一致

chinese_L-12_H-768_A-12为预训练的相关模型和词典

data文件夹中为训练语料,验证语料,测试语料

注意: 当语料中的分类种类发生变化时,run_classifier.py文件的SimProcessor中的get_labels方法也要随之变化

参数说明: max_seq_length sentence的最大长度(字) train_batch_size batch_size的大小