This repository contains a deep learning model based on Long Short-Term Memory (LSTM) networks designed for next word prediction. The model is capable of learning from a given text dataset and predicting the subsequent word in a sequence.
A trained model on "The Verdict (1908) by Edith Wharton" is deployed on Hugging Face, allowing for easy access.
You can access the model directly on Hugging Face by clicking here.
- Implements LSTM for next word prediction.
- Capable of training on any text dataset.
- Includes scripts for preprocessing, training, and prediction.
- Python
- TensorFlow
- NumPy
- Pandas
Clone the repository:
git clone https://github.com/Satvik-jain/DeepLearning_LSTM_WordPredictor.git
cd DeepLearning_LSTM_WordPredictor