movie-genre-classification

Movie genre classification using movie subtitles

Movie genre classification from movie subtitles is performed using NLP methods combined with ANN for 8 different movie genres.

GENRES: action - comedy - crime horror - musical - romance - war - western

DATA: https://drive.google.com/uc?id=1ciBIQiH6foHanRrDQKr-_9MBWfOKVc9d&export=download

Usage

$python2 predict_genre.py input-subtitle-path

Example

$cd src
$python2 predict_genre.py ../SayMyName.txt
--> Predicted Genre: Crime

Test subtitle SayMyName.txt is taken from one of the episodes of the American TV series Breaking Bad.