Skip to content

BERT for IMDb movie reviews binary classification into Positive and Negative

Notifications You must be signed in to change notification settings

siddsh/Movie-Review-Classification

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

13 Commits
 
 
 
 
 
 

Repository files navigation

Movie-Review-Classification

BERT for IMDb movie reviews binary classification into Positive and Negative

Implemented using PyTorch and the HuggingFace library trained on Google Colab.

Used BERT Base Model to get 94% accuracy.

Other metrics are
Benchmarks

We predicted the movie review of Parasite from Times of India Entertainment

Prediction Result

About

BERT for IMDb movie reviews binary classification into Positive and Negative

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Contributors 2

  •  
  •