SnowNLP实现情感分类

输入一个csv数据,把数据分类到积极,消极的csv文件中

# -*- coding:utf-8 -*-

from snownlp import SnowNLP
# 加载情感分析模块
from snownlp import sentiment
fileee="D:\PycharmProjects2020\qingganfenlei\data\weibo1.csv"
# text = '大麦多开一个口ok????正在现场俩口闲死 ​​​​'  # 文本

from snownlp import SnowNLP
import codecs
import os

f1 = open('./pos1.csv', 'a+',encoding='utf-8')  # 存放正面  名字也可自定义哦
f2 = open('./neg2.csv', 'a+',encoding='utf-8')  # 存放负面

source = open(fileee,"r")
line = source.readlines()

for i in line:
    if SnowNLP(i).sentiments < 0.3:  # 可以自定义范围
        print(i+'这是一个负面评价')
        print(SnowNLP(i).sentiments)
        # 这段文本写入neg文件中
        f2.write(i)
        # f2.write('\n')

    elif SnowNLP(i).sentiments > 0.8:  # 可以自定义范围
        print(i+'这是一个正面评价')
        print(SnowNLP(i).sentiments)
        # 这段文本写入pos文件中
        f1.write(i)
        # f1.write('\n')
    else:
        print('这是一个中性评价')

你可能感兴趣的:(nlp自然语言处理)