I try to create my own corpus for sentiment analysis of tweets (whether they are positive or negative).
I'm first trying the existing NLTK movie-review corpus. However, if I'm using this code:
import string
from itertools import chain
from nltk.corpus import movie_reviews as mr
from nltk.corpus import stopwords
from nltk.probability import FreqDist
from nltk.classify import NaiveBayesClassifier as nbc
import nltk
stop = stopwords.words('english')
documents = [([w for w in mr.words(i) if w.lower() not in stop and w.lower() not in string.punctuation], i.split('/')[0]) for i in mr.fileids()]
word_features = FreqDist(chain(*[i for i,j in documents]))
word_features = word_features.keys()[:100]
numtrain = int(len(documents) * 90 / 100)
train_set = [({i:(i in tokens) for i in word_features}, tag) for tokens,tag in documents[:numtrain]]
test_set = [({i:(i in tokens) for i in word_features}, tag) for tokens,tag in documents[numtrain:]]
classifier = nbc.train(train_set)
print nltk.classify.accuracy(classifier, test_set)
classifier.show_most_informative_features(5)
Im receiving output:
0.31
Most Informative Features
uplifting = True pos : neg = 5.9 : 1.0
wednesday = True pos : neg = 3.7 : 1.0
controversy = True pos : neg = 3.4 : 1.0
shocks = True pos : neg = 3.0 : 1.0
catchy = True pos : neg = 2.6 : 1.0
Instead of the expected output (see Classification using movie review corpus in NLTK/Python ):
0.655
Most Informative Features
bad = True neg : pos = 2.0 : 1.0
script = True neg : pos = 1.5 : 1.0
world = True pos : neg = 1.5 : 1.0
nothing = True neg : pos = 1.5 : 1.0
bad = False pos : neg = 1.5 : 1.0
I'm using exactly the same code as in the other StackOverflow page, my NLTK (and theirs) is up to date and I also have the most recent movie-reviews corpus. Anyone with an idea what's going wrong?
Thanks!