Python >> Tutoriel Python >  >> Python Tag >> NLTK

Est-il possible de modifier le lexique des sentiments vader de NLTK ?

Pour toute personne intéressée, cela peut également être réalisé sans avoir à modifier manuellement le fichier vader lexicon .txt. Une fois chargé, le lexique est un dictionnaire normal avec des mots comme clés et des scores comme valeurs. Tel que fourni par repoleved dans cet article :

from nltk.sentiment.vader import SentimentIntensityAnalyzer

new_words = {
    'foo': 2.0,
    'bar': -3.4,
}

SIA = SentimentIntensityAnalyzer()

SIA.lexicon.update(new_words)

Si vous souhaitez supprimer des mots, utilisez la fonction '.pop' :

SIA = SentimentIntensityAnalyzer()

SIA.lexicon.pop('no')

J'ai trouvé la solution. J'ai compressé le dossier vader_lexicon qui contient le fichier txt et les modifications que j'ai appliquées sont maintenant celles auxquelles on accède.