Fitting a model to a training dataset is so easy today with libraries like scikit-learn.
A model can be fit and evaluated on a dataset in just a few lines of code.
It is so easy that it has become a problem.
The same few lines of code are repeated again and again and it may not be obvious how to actually use the model to make a prediction.
Or, if a prediction is made, how to relate the predicted values to the actual input values.
I know that this is the case because I get many emails with the question:How do I connect the predicted values with the input data?This a common problem.
In this tutorial, you will discover how to relate the predicted values with the inputs to a machine learning model.
After completing this tutorial, you will know:Let’s get started.
How to Connect Model Input Data With Predictions for Machine LearningPhoto by Ian D.
Keating, some rights reserved.
This tutorial is divided into three parts; they are:Let’s start off by defining a dataset that we can use with our model.
You may have your own dataset in a CSV file or in a NumPy array in memory.
In this case, we will use a simple two-class or binary classification problem with two numerical input variables.
We can use the make_blobs() scikit-learn function to create this dataset with 1,000 examples.
The example below creates the dataset with separate arrays for the input (X) and outputs (y).
Running the example creates the dataset and prints the shape of each of the arrays.
We can see that there are 1,000 rows for the 1,000 samples in the dataset.
We can also see that the input data has two columns for the two input variables and that the output array is one long array of class labels for each of the rows in the input data.
Next, we will fit a model on this training dataset.
Now that we have a training dataset, we can fit a model on the data.
This means that we will provide all of the training data to a learning algorithm and let the learning algorithm to discover the mapping between the inputs and the output class label that minimizes the prediction error.
In this case, because it is a two-class problem, we will try the logistic regression classification algorithm.
This can be achieved using the LogisticRegression class from scikit-learn.
First, the model must be defined with any specific configuration we require.
In this case, we will use the efficient ‘lbfgs‘ solver.
Next, the model is fit on the training dataset by calling the fit() function and passing in the training dataset.
Finally, we can evaluate the model by first using it to make predictions on the training dataset by calling predict() and then comparing the predictions to the expected class labels and calculating the accuracy.
The complete example is listed below.
Running the example fits the model on the training dataset and then prints the classification accuracy.
In this case, we can see that the model has a 100% classification accuracy on the training dataset.
Now that we know how to fit and evaluate a model on the training dataset, let’s get to the root of the question.
How do you connect inputs of the model to the outputs?A fit machine learning model takes inputs and makes a prediction.
This could be one row of data at a time; for example:This is straightforward with our model.
For example, we can make a prediction with an array input and get one output and we know that the two are directly connected.
The input must be defined as an array of numbers, specifically 1 row with 2 columns.
We can achieve this by defining the example as a list of rows with a list of columns for each row; for example:We can then provide this as input to the model and make a prediction.
Tying this together with fitting the model from the previous section, the complete example is listed below.
Running the example defines the new input and makes a prediction, then prints both the input and the output.
We can see that in this case, the model predicts class label 1 for the inputs.
If we were using the model in our own application, this usage of the model would allow us to directly relate the inputs and outputs for each prediction made.
If we needed to replace the labels 0 and 1 with something meaningful like “spam” and “not spam“, we could do that with a simple if-statement.
So far so good.
What happens when the model is used to make multiple predictions at once?That is, how do we relate the predictions to the inputs when multiple rows or multiple samples are provided to the model at once?For example, we could make a prediction for each of the 1,000 examples in the training dataset as we did in the previous section when evaluating the model.
In this case, the model would make 1,000 distinct predictions and return an array of 1,000 integer values.
One prediction for each of the 1,000 input rows of data.
Importantly, the order of the predictions in the output array matches the order of rows provided as input to the model when making a prediction.
This means that the input row at index 0 matches the prediction at index 0; the same is true for index 1, index 2, all the way to index 999.
Therefore, we can relate the inputs and outputs directly based on their index, with the knowledge that the order is preserved when making a prediction on many rows of inputs.
Let’s make this concrete with an example.
First, we can make a prediction for each row of input in the training dataset:We can then step through the indexes and access the input and the predicted output for each.
This shows precisely how to connect the predictions with the input rows.
For example, the input at row 0 and the prediction at index 0:In this case, we will just look at the first 10 rows and their predictions.
Tying this together, the complete example of making a prediction for each row in the training data and connecting the predictions with the inputs is listed below.
Running the example, the model makes 1,000 predictions for the 1,000 rows in the training dataset, then connects the inputs to the predicted values for the first 10 examples.
This provides a template that you can use and adapt for your own predictive modeling projects to connect predictions to the input rows via their row index.
This section provides more resources on the topic if you are looking to go deeper.
In this tutorial, you discovered how to relate the predicted values with the inputs to a machine learning model.
Specifically, you learned:Do you have any questions?.Ask your questions in the comments below and I will do my best to answer.
with just a few lines of scikit-learn codeLearn how in my new Ebook: Machine Learning Mastery With PythonCovers self-study tutorials and end-to-end projects like: Loading data, visualization, modeling, tuning, and much more.
Skip the Academics.
Just Results.
.