这是一个基于bert的中文文本情感分类模型,可以通过命令行输入文本进行基本的文本情感分析
数据集是带情感标签的微博评论对话数据集,总共有8414条对话数据,每条包含4到6句对话语句。标签1代表happiness,标签2代表Love,标签3代表Sorrow,标签4代表Fear,标签5代表Disgust,标签6代表None
由于无法上传模型参数,需要先训练再使用
- 首先运行train.py训练模型,模型参数,参数会自动保存在工程目录下
- 运行run.py,加载训练好的模型参数,在对话框输入文本就可以看到模型分类结果