BruceJust / Sentiment-classification-by-BERT

Sentiment classification by BERT on own data

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

Sentiment-classification-by-BERT

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模型比较强大,一上来就容易达到很高分了。

img

About

Sentiment classification by BERT on own data


Languages

Language:Python 100.0%