ebm_show: ebm_show

In interpret: Fit Interpretable Machine Learning Models

ebm_show

Description

Shows the GAM plot for a single feature

Usage

ebm_show(
  model, 
  name
)

Arguments

model	the model
name	the name of the feature to plot

Value

None

Examples

  data(mtcars)
  X <- subset(mtcars, select = -c(vs))
  y <- mtcars$vs

  set.seed(42)
  data_sample <- sample(length(y), length(y) * 0.8)

  X_train <- X[data_sample, ]
  y_train <- y[data_sample]
  X_test <- X[-data_sample, ]
  y_test <- y[-data_sample]

  ebm <- ebm_classify(X_train, y_train)
  ebm_show(ebm, "mpg")

interpret documentation built on Feb. 16, 2023, 8:07 p.m.