View source: R/plot_learner_prediction.R
plot_learner_prediction | R Documentation |
Generates a plot for the mlr3::Prediction of a single mlr3::Learner on a single mlr3::Task.
For classification we support tasks with exactly two features and learners
with predict_type
set to "response"
or "prob"
.
For regression we support tasks with one or two features.
For tasks with one feature we print confidence bounds if the predict type
of the learner was set to "se"
.
For tasks with two features the predict type will be ignored.
Note that this function is a wrapper around autoplot.ResampleResult()
for a
temporary mlr3::ResampleResult using mlr3::mlr_resamplings_holdout with
ratio 1 (all observations in training set).
plot_learner_prediction(learner, task, grid_points = 100L, expand_range = 0)
learner |
(mlr3::Learner). |
task |
(mlr3::Task). |
grid_points |
( |
expand_range |
( |
ggplot2::ggplot()
object.
if (requireNamespace("mlr3")) { library(mlr3) library(mlr3viz) task = mlr3::tsk("pima")$select(c("age", "glucose")) learner = lrn("classif.rpart", predict_type = "prob") p = plot_learner_prediction(learner, task) print(p) }
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.