Skip to content

Latest commit

 

History

History
25 lines (19 loc) · 1.38 KB

README.md

File metadata and controls

25 lines (19 loc) · 1.38 KB

LSTM_Sentiment_Classification

Implementation of LSTM architecture in Keras to perform sentiment analysis on movie reviews from the Large Movie Review Dataset.

Dataset

The Large Movie Review Dataset (often referred to as the IMDB dataset) contains 25,000 highly-polar movie reviews (good or bad) for training and the same amount again for testing. The problem is to determine whether a given movie review has a positive or negative sentiment.

Link to download dataset here

The data was collected by Stanford researchers and was used in a 2011 paper where a split of 50-50 of the data was used for training and test. An accuracy of 88.89% was achieved.

Models

  1. A Simple LSTM Network
  2. LSTM Network With Dropout
  3. LSTM and Convolutional Neural Network with dropout

Results

  1. Simple LSTM - 85.72%
  2. LSTM Network With Dropout - 85.67%
  3. LSTM and Convolutional Neural Network with dropout - 88.41% !

Credits -

The following resources are extremely useful for a detailed understanding of this task or Deep Learning techniques:

  1. WildML blog by Denny Britz
  2. Machine Learning Mastery blog by Dr.Jason Brownlee
  3. Andrej Karpathy's blog on RNNs