plotHeatmap: Heatmap

In Pizzaknoedel/visualize-hyperparameter: Visualizing Hyperparamer Performance Dependencies

Description

Visualization of a heatmap from a mlr3::Task object.

Usage

plotHeatmap(
  task,
  features = NULL,
  fun = NULL,
  gridsize = 20,
  scatterplot = FALSE,
  rug = TRUE,
  title = TRUE,
  numericNA = "Max"
)

Arguments

task	(mlr3::Task object) A task for which the plot should be generated.
features	(character(2)) The names of the two columns to be plotted. If NULL, the first two feature columns will be plotted. The default is NULL.
fun	(function) A summary function used in the rectangles of the grid. The default is mean for a regression task and majority vote for a classification task.
gridsize	(numeric(1)) The number of rectangles per axis. The default value is 20.
scatterplot	(logical) If TRUE, a scatterplot will be plotted. The default is FALSE.
rug	(logical) If TRUE, a rug will be plotted. The default is TRUE.
title	(logical) If TRUE, the title will be plotted. The default is TRUE.
numericNA	(character(1) If "Max", the NA values of the column are displayed as maximum value. If "Min", the NA values are displayed as minimum value. The default is "Max".

Value

A plotly object.

See Also

plotParallelCoordinate plotPartialDependence plotImportance

Examples

library(mlr3)
data(glmnet_ela)
task_glmnet_ela = TaskRegr$new(id = "task_glmnet", backend = glmnet_ela, target = "logloss")
plotHeatmap(task_glmnet_ela)

Pizzaknoedel/visualize-hyperparameter documentation built on Feb. 13, 2022, 8:11 a.m.