使用 BertForSequenceClassification
,在自己的训练集上训练情感分类模型
数据集来源于 https://github.com/bojone/bert4keras/tree/master/examples/datasets
是一个中文的情感二分类数据集。
词汇表 vocab.txt
来自于哈工大的中文预训练语言模型 BERT-wwm, Chinese
。
BertPractice.py
:使用 BertForSequenceClassification
,训练情感分类模型
代码讲解,请参考我的博客文章:Bert 文本分类实战