基于微调BERT的中文评论情感分类模型(支持CPU、GPU)
基于微调BERT的中文评论情感分类模型采用*ktrain*库进行文本分类0.分配GPU(CPU版略过)1.加载数据2.打乱数据3.预处理数据并构造一个中文trasnformer模型估计一个较好的学习率训练模型评估模型在新的数据上进行测试找到loss最大的样本在新的数据上进行测试保存模型,以便之后调用结果分析实验总结迁移中文bert模型(chinese-bert-wwm),完成对20000余条评论数据的情感分类,分为0-很好,1-一般,2-较差。学习到bert模型的强大之后,不禁感叹,之前比赛用的模型太垃
原创
2021-12-15 13:27:30 ·
4349 阅读 ·
0 评论