ebm_predict_proba {interpret}

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)