Next Word Prediction with Python

You must therefore have seen the prediction of the next word on the keyboard of our smartphones which is most of the time precise. How does our keyboard predict the next word? In this article, I will walk you through the task of the next word prediction with Python.

How To Predict the Next Word?

Predicting the next word is a machine learning task, but we can also achieve good results using only Python. Below are the steps you need to take when writing code to predict the next word with Python:

First, divide the sentence into words
Then select the last word of the sentence
Then find the probability of the last word by looking at the vocabulary (dataset).
Then, finally, select the next possible word.

In the section below, I will walk you through the task of the next word prediction with Python using all the steps mentioned above.

Also, Read – 100+ Machine Learning Projects Solved and Explained.

Next Word Prediction with Python

In this section, I’ll walk you through a simple Python code to predict the next word using the Python programming language. So, to start with the task of next word prediction with Python, you need to have a vocabulary as shown below.

what time is it?
what is it?
what is time?
is it time?
Note: Kind of data you need in a text file.

Now, I will start by defining a Python function that will help to fill the vocabulary of words the right way by counting the frequency and combination of words:

So we created the function to fill the vocabulary of words, now let’s use this function on the dataset which will just split the sentences into words and select the last word of the sentence:

Now we are ready to find the next predicted words:

The code above will take user input and find your input in the dataset, so to test your code, use the sentences that are familiar with the sentences written by you in the dataset. The code will find the next word and print it at the end of your input sentence.

This is how we can predict the next word with Python. Below is the complete code used in this article for the next word prediction task with Python.

I hope you liked this article on the task of Next word prediction with Python. Feel free to ask your valuable questions in the comments section below.