https://github.com/pengming617/bert_classification

利用bert预训练的中文模型进行文本分类 数据集中文情感分析语料chnsenticorp
https://github.com/pengming617/bert_classification

Last synced: about 1 month ago
JSON representation

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

• Host: GitHub
• URL: https://github.com/pengming617/bert_classification
• Owner: pengming617
• Created: 2018-12-21T08:20:07.000Z (almost 6 years ago)
• Default Branch: master
• Last Pushed: 2019-08-26T03:02:16.000Z (over 5 years ago)
• Last Synced: 2024-08-02T08:09:55.514Z (4 months ago)
• Language: Python
• Homepage:
• Size: 3.87 MB
• Stars: 302
• Watchers: 6
• Forks: 73
• Open Issues: 6
• Metadata Files:
  • Readme: README.md

Awesome Lists containing this project

• awesome-bert - pengming617/bert_classification

README

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