And finally, we make predictions on the test data, using the trained model: