
Plots for Rpart Learners
Source:R/LearnerClassifRpart.R
, R/LearnerRegrRpart.R
autoplot.LearnerClassifRpart.Rd
Visualizations rpart trees using the package ggparty.
Arguments
- object
- theme
(
ggplot2::theme()
)
Theggplot2::theme_minimal()
is applied by default to all plots.- ...
(ignored).
Examples
if (requireNamespace("mlr3")) {
library(mlr3)
library(mlr3viz)
# classification
task = tsk("iris")
learner = lrn("classif.rpart", keep_model = TRUE)
learner$train(task)
autoplot(learner)
# regression
task = tsk("mtcars")
learner = lrn("regr.rpart", keep_model = TRUE)
learner$train(task)
autoplot(learner)
}