tfidf_vader.py 文件源码

def vader(term):
    data = term[1]
    result = {'compound':[], 'neg':[], 'neu':[], 'pos':[] }
    sid = SentimentIntensityAnalyzer()
    for sentence in data:
        ss = sid.polarity_scores(sentence)
        result['compound'].append(ss['compound'])
        result['neg'].append(ss['neg'])
        result['neu'].append(ss['neu'])
        result['pos'].append(ss['pos'])
    vaderList = [sum(result[i]) for i in result.keys()]
    list = [term[0],vaderList]
    resultsKeys = result.keys() 
    db_data = ({'article': list[0], 'negative': list[1][0], 'neutral': list[1][1], 'positive': list[1][2], 'compound': list[1][3]})
    try:
        with db.atomic():
                ArticleFeature.insert(db_data).execute()
    except peewee.IntegrityError:
        print term[0]
        #print 'Skipping Duplicate'