ebm_predict_proba: ebm_predict_proba

In interpret: Fit Interpretable Machine Learning Models

ebm_predict_proba

Description

Predicts probabilities using an EBM model

Usage

ebm_predict_proba(
  model, 
  X
)

Arguments

model	the model
X	features

Value

returns the probabilities predicted

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)
  proba_test <- ebm_predict_proba(ebm, X_test)

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