Skip to content

pengming617/bert_classification

Repository files navigation

利用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的大小

About

利用bert预训练的中文模型进行文本分类 数据集中文情感分析语料chnsenticorp

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published