--- Large Data Analysis Course Project ---
This folder is a set of simplified python codes which use sklearn package to classify movie reviews.
Two classifiers were used: Naive Bayes and SVM. SVM gives an accuracy of about 87.5%, which is slightly higher than 86% given by Naive Bayes.
imdbReviews.py generates *.pkl files which are the training and testing datasets.
First, set the dataset directory in the imdbReviews.py, then run the code.
python imdbReviews.pyYou will get two *.pkl files which are needed for naive.py and svm.py.
To do prediction, run the following command.
python naive.py
last modified 04/16/2016