IMDB-Classifier

ECEN 489: Bag of words meets bag of popcorn

To predict if the review is positive or negative using Machine Learning algorithms.

Dataset: IMDB reviews

Instruction to run the code:

Before running the program, ensure to download the dataset from the given link and store it in the Data/ folder

1. For IMDBClassifier.py (Word2Vec + trigrams + Bi-LSTM)

   $ python ./IMDBClassifier.py

2. For ReviewClassifier.py (FC model)

   $ python ./ReviewClassifier.py