Skip to content

Latest commit

 

History

History
43 lines (27 loc) · 1.84 KB

README.md

File metadata and controls

43 lines (27 loc) · 1.84 KB

Multi-label Genre Prediction

A genre prediction model that given a plot allows the user to predict possible genres for a movie or tv-show.

Deployment

The app can be found at: predict-genre.streamlit.app

Motivation

This model can be used to analyze synopses and identify potential mislabeling or inconsistencies in genre assignment. This can contribute to maintaining the quality and accuracy of genre metadata in media databases.

image

CleanShot 2023-12-21 at 07 53 53@2x

ML Pipeline

Underlying Model - BiLSTM

Given a huge description, Bidirectional LSTM (BiLSTM) with the ability to process input flows in both directions can significantly help with NLP and genre context development in our case

CleanShot 2023-12-21 at 07 57 55@2x

Nerual Architecture

  • Input Neurons = 350 / 500
  • Embedding Size = 300
  • BiLSTM L1 Size = 128
  • BiLSTM L2 Size = 64
  • Dense Layer 1 Neurons = 64
  • Dense Layer 2 Neurons = 27

Screenshots

CleanShot 2023-12-21 at 08 05 45@2x

CleanShot 2023-12-21 at 08 06 03@2x

CleanShot 2023-12-21 at 08 06 42@2x

Collaborators / Team Members