Skip to content

Classifies the movie reviews as positive or negative using LSTM Networks

Notifications You must be signed in to change notification settings

Surya-Prakash-Reddy/Movie-Review-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

Movie Review Classification - Sentiment Analysis

Classifies whether a movie review is positive or negative using LSTM Networks.

Table of contents

About Project:

The project is about classification of reviews of movies as positive or negative. The dataset is taken from the IMDB and has labels as positive or negative. The model is built using LSTM and has got accuracy over 84% with the test data.

Languages or Frameworks Used

  • Python: language
  • NumPy: library for numerical calculations
  • Matplotlib: library for data visualisation
  • PIL: Python Image Library for opening and manage different image formats
  • Pytorch: a deep learning framework by Facebook AI Research Team for building neural networks
  • torchvision: package consists of popular datasets, model architectures, and common image transformations for computer vision

Setup

To use this project, clone the repo

Clone

  git clone https://github.com/Surya-Prakash-Reddy/Movie-Review-Classification.git

After cloning, you can use the Sentiment Cloud.ipynb or Untitled.ipynb notebook to learn or modify. Sentiment Cloud.ipynb is modified from Untitled.ipynb to train on Google Colab.