Modern Scala Projects

Running the LR model on the test dataset

We will now invoke the transform method on our LogisticRegression model:

//Next run the model on the test dataset and obtain predictionsval testDataPredictions = logitModel.transform(testDataSet)

The transform method invocation returns a new DataFrame. Not only that, that model transformation step resulted in three new columns:

• rawPrediction
• probability
• predictions

Next, we will display the first 25 rows of this DataFrame: 

testDataPredictions.show(25)

See the following table to look at the displayed predictions, which were made by running our LR models on the test dataset:

Three new columns resulting ...