Skip to content

Repository files navigation

Next Word Predictor: Hamlet LSTM-RNN

This project is a deep learning-based web application that predicts the next word in a sequence of text. The model is a Long Short-Term Memory (LSTM) neural network trained on the text of William Shakespeare's Hamlet.

🛠️ Features

  • Deep Learning Model: Utilizes an LSTM-RNN architecture for sequence prediction.
  • Natural Language Processing: Employs word tokenization and padding to process input text.
  • Interactive Interface: A user-friendly UI built with Streamlit for real-time word generation.
  • Early Stopping: The model was trained with early stopping to prevent overfitting and ensure better generalization.

Tech Stack

Installation & Setup

To run this project locally, follow these steps:

  1. Clone the repository:

    git clone https://github.com/SidoJain/Next-Word-Prediction-LSTM-RNN.git
  2. Create a virtual environment:

    python -m venv .venv
    source .venv/Script/activate
  3. Install dependencies:

    pip install -r requirements.txt
  4. Launch the app:

    streamlit run app.py

About

This project is a deep learning-based web application that predicts the next word in a sequence of text. The model is a Long Short-Term Memory (LSTM) neural network trained on the text of William Shakespeare's Hamlet.

Topics

Resources

Stars

0 stars

Watchers

0 watching

Forks

Releases

Packages

Contributors

Languages