Learn R Programming

fastml (version 0.7.0)

plot_ice: Plot ICE curves for a fastml model

Description

Generates Individual Conditional Expectation (ICE) plots for selected features using the `pdp` package.

Usage

plot_ice(object, features, ...)

Value

A `ggplot` object displaying ICE curves.

Arguments

object

A `fastml` object.

features

Character vector of feature names to plot.

...

Additional arguments passed to `pdp::partial`.

Examples

Run this code
if (FALSE) {
data(iris)
iris <- iris[iris$Species != "setosa", ]
iris$Species <- factor(iris$Species)
model <- fastml(data = iris, label = "Species")
plot_ice(model, features = "Sepal.Length")
}

Run the code above in your browser using DataLab