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.