explore (version 1.3.4)

predict_target: Predict target using a trained model.

Description

Predict target using a trained model.

Usage

predict_target(data, model, name = "prediction")

Value

data containing predicted probabilities for target values

Arguments

data

A dataset (data.frame or tbl)

model

A model created with explain_*() function

name

Prefix of variable-name for prediction

Examples

if (FALSE) { # rlang::is_installed(c("rpart", "randomForest", "xgboost"))
data_train <- create_data_buy(seed = 1)
data_test <- create_data_buy(seed = 2)
model <- explain_tree(data_train, target = buy, out = "model")
data <- predict_target(data = data_test, model = model)
describe(data)
}