Data Science Desktop Survival Guide
by Graham Williams |
|||||
Enhanced Plots: Add Percentage of Observations |
prp(model, type=2, extra=106)
This is a particularly informative plot for binary classification tasks. Each node includes the probability of the second class, which is usually the positive class in a binary classification dataset. Each node also includes the percentage of the training dataset that corresponds to that node.
|