mlr3viz (version 0.5.3)

autoplot.LearnerClustHierarchical: Plot for Hierarchical Clustering Learners

Description

Visualize dendrograms for hierarchical clusterers using the package factoextra.

Note that learner-specific plots are experimental and subject to change.

Usage

# S3 method for LearnerClustHierarchical
autoplot(object, ...)

Arguments

object

(mlr3cluster::LearnerClustAgnes | mlr3cluster::LearnerClustDiana).

...

(any): Additional arguments, passed down to function factoextra::fviz_dend() in package factoextra.

Value

ggplot2::ggplot() object.

Examples

# NOT RUN {
library(mlr3)
library(mlr3cluster)
library(mlr3viz)

task = mlr_tasks$get("usarrests")

# agnes clustering
learner = mlr_learners$get("clust.agnes")
learner$train(task)
autoplot(learner)

# diana clustering
learner = mlr_learners$get("clust.diana")
learner$train(task)
autoplot(learner,
  k = learner$param_set$values$k, rect_fill = TRUE,
  rect = TRUE, rect_border = "red")
# }