Sentiment classification by BERT on own data
基于自己的数据,使用BERT fine-tuning
该项目使用的外卖的情感分析数据,也在自己的数据上试过,效果不错,供参考。
安装包:
1.transformers
2.tensorflow
3.pytorch(可以不要)
本案例基于huggingface出品的transformers进行BERT fine-tuning。
主要修改glue.py及runtf_glue.py来完成。
详细步骤我在blog写了:
https://blog.csdn.net/weixin_42175217/article/details/105476646
训练结果:
10个epoch
可以看到train_acc一直在增长,val_acc没怎么变,因为已经到88,89了。
BERT模型比较强大,一上来就容易达到很高分了。