SnowNLP文本情感分析

文本情感分析,可以自定义词典、训练模型来进行,但花费时间长。如果采用现有的模型,直接用几行代码就可以得出文本的情感倾向,现有的模型主要有SnowNLP和Cemotion。

SnowNLP用的是经典机器学习的贝叶斯算法;

Cemotion使用的是深度学习RNN模型,目前Cemotion使用的模型为rnn_emotion_x86_1.0.h5。

完整代码(SnowNLP,需要pip 安装):

from snownlp import SnowNLP
import pandas as pd

def get_reviews(file_path):
    predict = []
    data = pd.read_csv(file_path,sep="\t")
    review = data.iloc[:,1]
    emotion = [SnowNLP(str(a)) for a in list(review)]
    return emotion


file_path = 'd:/datasets/reviews.txt'
emotions = get_reviews(file_path)
for emotion in emotions:
  print('"', ",".join(emotion.sentences) , '"\n' , '情感倾向:{:6f}'.format(emotion.sentiments) , '\n')
" 看到朋友的姐姐用这个说效果很好,我也买来给姐姐和妈妈用,都说效果不错,皮肤好多了,又白又嫩~点个赞 "
 情感倾向:0.999875 

" 质量很好,绝对的正品,包装也很好,没有任何损坏,非常满意 "
 情感倾向:0.951805 

" 口罩收到了质量很好,包装严实,透气性很好 "
 情感倾向:0.862890 

" 质量很好疫情期间必不可少的便宜有活动多囤点 "
 情感倾向:0.918779 

" 是正品,干净卫生,做过好独立包装,赞 "
 情感倾向:0.994066 

" 太差了,用了一次就坏了 "
 情感倾向:0.004719 

" 不好用,下次不会买了 "
 情感倾向:0.396294 

" 不会回购了 "
 情感倾向:0.370883 

测试集reviews.txt(注:id与review之间tab键分隔)

id    review
1    看到朋友的姐姐用这个说效果很好,我也买来给姐姐和妈妈用,都说效果不错,皮肤好多了,又白又嫩~点个赞!
2    质量很好,绝对的正品,包装也很好,没有任何损坏,非常满意。
3    口罩收到了质量很好,包装严实,透气性很好
4    质量很好疫情期间必不可少的便宜有活动多囤点
5    是正品,干净卫生,做过好独立包装,赞
6    太差了,用了一次就坏了
7    不好用,下次不会买了
8    不会回购了

你可能感兴趣的:(深度学习,神经网络,机器学习,深度学习,人工智能,情感分析)