Skip to content

This repo contains a Jupyter notebook for training an English next word prediction model based on a conversational dataset.

Notifications You must be signed in to change notification settings

arusl/nwp-en-bsd

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

4 Commits
 
 
 
 

Repository files navigation

Next Word Prediction Model

This notebook shows how to train an English next word prediction model using LSTM (w/ Keras and NLTK).

Required files

To run the notebook (next_word_prediction_en_bsd.ipynb) in Google Colab, you need to download the followings and adjust the paths inside the notebook:

Load a trained model

If you want, you can skip the training step and load the trained model below to make predictions.

Notes

Detailed explanations can be found inside the notebook.

About

This repo contains a Jupyter notebook for training an English next word prediction model based on a conversational dataset.

Topics

Resources

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published