Route rows down a fitted CART tree (morie.fn hmcart::predict_tree)

Route rows down a fitted CART tree (morie.fn hmcart::predict_tree)

Usage

morie_geron_predict_tree(tree, X)

Arguments

tree

Tree from morie_geron_cart_algorithm.

X

Rows to route.

Value

Vector of leaf values, one per row.

Examples

set.seed(5)
X <- matrix(rnorm(60), 30, 2)
y <- as.integer(X[, 1] > 0)
fit <- morie_geron_cart_algorithm(X, y)
morie_geron_predict_tree(fit$tree, X[1:5, , drop = FALSE])
#> [1] 0 1 0 1 1