Skip to content

Sanjaypranav/movie-genre-text-classification

Repository files navigation

Movie-Genre-classification

Project Description

To Classify Genre of the movie from the given Overview

Abstract

This model is Trained with Disney+ Hotstar Dataset it works well for anime, cartoon and action related movies

Dataset Description

Dataset collected from Kaggle, For further data you can choose Hotstar dataset but model will take time to train.

Algorithm used

    OnevsRest Classifier
    Naive Bayes

App deployment

FLASK APP

To run this Project locally

Step 1: Create conda environment

conda create -n env_name python=3.7 -y

Linux users

virtual env python=python3.9 env_name

Step 2: clone the repo

git clone https://github.com/Sanjaypranav/movie-genre-text-classification.git

step 3 :install the requirements

pip install -r requirements.txt

step 4 : run the app.py file

python app.py