1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
106
107
108
109
|
% Generated by roxygen2: do not edit by hand
% Please edit documentation in R/plotLearnerPrediction.R
\name{plotLearnerPrediction}
\alias{plotLearnerPrediction}
\title{Visualizes a learning algorithm on a 1D or 2D data set.}
\usage{
plotLearnerPrediction(
learner,
task,
features = NULL,
measures,
cv = 10L,
...,
gridsize,
pointsize = 2,
prob.alpha = TRUE,
se.band = TRUE,
err.mark = "train",
bg.cols = c("darkblue", "green", "darkred"),
err.col = "white",
err.size = pointsize,
greyscale = FALSE,
pretty.names = TRUE
)
}
\arguments{
\item{learner}{(\link{Learner} | \code{character(1)})\cr
The learner.
If you pass a string the learner will be created via \link{makeLearner}.}
\item{task}{(\link{Task})\cr
The task.}
\item{features}{(\link{character})\cr
Selected features for model.
By default the first 2 features are used.}
\item{measures}{(\link{Measure} | list of \link{Measure})\cr
Performance measure(s) to evaluate.
Default is the default measure for the task, see here \link{getDefaultMeasure}.}
\item{cv}{(\code{integer(1)})\cr
Do cross-validation and display in plot title?
Number of folds. 0 means no CV.
Default is 10.}
\item{...}{(any)\cr
Parameters for \code{learner}.}
\item{gridsize}{(\code{integer(1)})\cr
Grid resolution per axis for background predictions.
Default is 500 for 1D and 100 for 2D.}
\item{pointsize}{(\code{numeric(1)})\cr
Pointsize for ggplot2 \link[ggplot2:geom_point]{ggplot2::geom_point} for data points.
Default is 2.}
\item{prob.alpha}{(\code{logical(1)})\cr
For classification: Set alpha value of background to probability for
predicted class? Allows visualization of \dQuote{confidence} for prediction.
If not, only a constant color is displayed in the background for the predicted label.
Default is \code{TRUE}.}
\item{se.band}{(\code{logical(1)})\cr
For regression in 1D: Show band for standard error estimation?
Default is \code{TRUE}.}
\item{err.mark}{(\code{character(1)}):
For classification: Either mark error of the model on the training data (\dQuote{train}) or
during cross-validation (\dQuote{cv}) or not at all with \dQuote{none}.
Default is \dQuote{train}.}
\item{bg.cols}{(\code{character(3)})\cr
Background colors for classification and regression.
Sorted from low, medium to high.
Default is \code{TRUE}.}
\item{err.col}{(\code{character(1)})\cr
For classification: Color of misclassified data points.
Default is \dQuote{white}}
\item{err.size}{(\code{integer(1)})\cr
For classification: Size of misclassified data points.
Default is \code{pointsize}.}
\item{greyscale}{(\code{logical(1)})\cr
Should the plot be greyscale completely?
Default is \code{FALSE}.}
\item{pretty.names}{(\code{logical(1)})\cr
Whether to use the short name of the learner instead of its ID in labels. Defaults to \code{TRUE}.}
}
\value{
The ggplot2 object.
}
\description{
Trains the model for 1 or 2 selected features, then displays it via \link[ggplot2:ggplot]{ggplot2::ggplot}.
Good for teaching or exploring models.
For classification and clustering, only 2D plots are supported. The data points, the classification and
potentially through color alpha blending the posterior probabilities are shown.
For regression, 1D and 2D plots are supported. 1D shows the data, the estimated mean and potentially
the estimated standard error. 2D does not show estimated standard error,
but only the estimated mean via background color.
The plot title displays the model id, its parameters, the training performance
and the cross-validation performance.
}
|
