plot_partial_dependency: Generate a partial dependency plot

In breather/brightbox: Peek into any blackbox learner (including ensembles)

Description

Take the output from calculate_partial_dependency and plot the partial dependencies for every model. This function returns the plot object so it can be further modified.

Usage

plot_partial_dependency(pd)

Arguments

pd	output from calculate_partial_dependency

Examples

## Not run: 
# Example output from calculate_partial_dependency
pd <- data.table(feature = rep("a", 9),
                 feature_val = rep(c(1, 3.5, 6), 3),
                 model = rep(c("model1",
                               "model2",
                               "ensemble"),
                             each = 3),
                 prediction = c(c(-2.5, 0, 2.5),
                                c(0, 0, 0),
                                c(-2.5, -0.75, 0)))
plot_partial_dependency(pd)

## End(Not run)

breather/brightbox documentation built on May 13, 2019, 5:04 a.m.