ebm_predict_proba | R Documentation |
Predicts probabilities using an EBM model
ebm_predict_proba(
model,
X
)
model |
the model |
X |
features |
returns the probabilities predicted
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)
Add the following code to your website.
For more information on customizing the embed code, read Embedding Snippets.