Skip to content

Latest commit

 

History

History
36 lines (18 loc) · 636 Bytes

README.md

File metadata and controls

36 lines (18 loc) · 636 Bytes

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