Machine Learning with R Cookbook - Second Edition

How it works...

In this recipe, we first demonstrate that one can use the prediction function to predict the category (class label) of the testing dataset testset, and then employ a table function to generate a classification table. Next, you can use the confusionMatrix function built into the caret package to determine the performance measurements.

In addition to the predict function, treeresponse is also capable of estimating the class probability, which will often classify labels with a higher probability. In this example, we demonstrated how to obtain the estimated class probability using the top five records of the testing dataset testset. The treeresponse function returns a list of five probabilities. You can use the list to determine ...