unimore/bda-6-steffo
unimore/bda-6-steffo
1
Fork 0
mirror of https://github.com/Steffo99/unimore-bda-6.git synced 2024-10-16 14:27:32 +00:00
Code Activity

Fix access to ratings instead of tokens

Browse source
This commit is contained in:
Steffo 2023-02-13 15:43:46 +01:00
parent bf193bcd93
commit d4373f8c02
Signed by: steffo
GPG key ID: 2A24051445686895
1 changed files with 1 additions and 1 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

2
unimore_bda_6/analysis/nltk_sentiment.py
View file

@ -33,7 +33,7 @@ class NLTKSentimentAnalyzer(BaseSentimentAnalyzer):
Register the `nltk.sentiment.util.extract_unigram_feats` feature extrator on the model.
"""
# Ignore the category and only access the tokens
tokenbags = map(lambda r: r.rating, dataset)
tokenbags = map(lambda r: r.tokens, dataset)
# Get all words in the documents
all_words = self.model.all_words(tokenbags, labeled=False)
# Create unigram `contains(*)` features from the previously gathered words