Skip to content

BruceJust/Sentiment-classification-by-BERT

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

6 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages