This function is mostly useful in an educational setting. Can only be used with trained workflow objects with 2 numeric predictor variables.

viz_decision_boundary(x, new_data, resolution = 100, expand = 0.1)

Arguments

x

trained `workflows::workflow` object.

new_data

A data frame or tibble for whom the preprocessing will be applied.

resolution

Number of squared in grid. Defaults to 100.

expand

Expansion rate. Defaults to 0.1. This means that the width and height of the shaded area is 10 data.

The chart have been minimally modified to allow for easier styling.

Value

`ggplot2::ggplot` object

Examples

library(parsnip) library(workflows) svm_spec <- svm_rbf() %>% set_mode("classification") %>% set_engine("kernlab") svm_fit <- workflow() %>% add_formula(Species ~ Petal.Length + Petal.Width) %>% add_model(svm_spec) %>% fit(iris) viz_decision_boundary(svm_fit, iris)
viz_decision_boundary(svm_fit, iris, resolution = 20)
viz_decision_boundary(svm_fit, iris, expand = 1)
svm_multi_fit <- workflow() %>% add_formula(class ~ umap_1 + umap_2) %>% add_model(svm_spec) %>% fit(mnist_sample) viz_decision_boundary(svm_multi_fit, mnist_sample)