Skip to content

Generating movie plots based on genres πŸŽ₯πŸŽžοΈπŸŽ¬πŸ“½οΈ

Notifications You must be signed in to change notification settings

farzeennimran/AI-Movie-Plot-Generator

Folders and files

NameName
Last commit message
Last commit date

Latest commit

Β 

History

5 Commits
Β 
Β 
Β 
Β 

Repository files navigation

AI Movie Plot Generator

The purpose of AI plot generator is to create a smart model capable of generating movie plots based on their genres.

Dataset

Data was collected from IMDb and Kaggle which was then compiled into one dataset and is available on kaggle: https://www.kaggle.com/datasets/farzeenimran/movieplots/data

Preprocessing

  1. Lowercasing:

  2. Special Characters Removal:

  3. Stop Words Removal:

  4. Tokenization:

  5. Genre Mapping:

  6. Handling Null Plots:

  7. Dealing with Missing Values:

  8. Vectorization:

Models applied

  1. RNN

  2. LSTM with embedding layer

  3. Markov Model

  4. Transformers