Plot Method for the train Class
This function takes the output of a
train object and creates
a line or level plot using the
plot.train(x, plotType = "scatter", metric = c("Accuracy", "RMSE"), digits = getOption("digits") - 5, xTrans = NULL, ...)
- an object of class
- What measure of performance to plot. Possible values are "RMSE", "Rsquared", "Accuracy" or "Kappa"
- a string describing the type of plot (
- an integer specifying the number of significant digits used to label the parameter value.
- a fuction that will be used to scale the x-axis in scatter plots.
- specifications to be passed to
bwplot(for line plots). These values should not be axis labels, panel functions, or titles.
If there are no tuning parameters, or none were varied, a plot of the resampling distirbution is produced via
If the model has one tuning parameter with multiple candidate values, a
plot is produced showing the profile of the results over the
parameter. Also, a plot can be produced if there are multiple
tuning parameters but only one is varied.
If there are two tuning parameters with different values, a plot can be produced where a different line is shown for each value of of the other parameter. For three parameters, the same line plot is created within conditioning panels of the other parameter.
Also, with two tuning parameters (with different values), a levelplot (i.e. un-clustered heatmap) can be created. For more than two parameters, this plot is created inside conditioning panels.
data(iris) TrainData <- iris[,1:4] TrainClasses <- iris[,5] library(e1071) rpartFit <- train(TrainData, TrainClasses, "rpart", tuneLength=15) plot(rpartFit, scales = list(x = list(rot = 90))) library(klaR) rdaFit <- train(TrainData, TrainClasses, "rda", control = trainControl(method = "cv")) plot(rdaFit, plotType = "line", auto.key = TRUE)