predict_hebart: Predictions for the HEBART model

predict_hebartR Documentation

Predictions for the HEBART model

Description

This function predicts for the final trees of a Hierachical Embedded BART model

Usage

predict_hebart(model, newdata, formula, group_variable)

Arguments

model

The model object

newdata

The new data to predict

formula

The model formula

group_variable

The grouping variable

Value

A dataframe with the prediction column

Author(s)

Bruna Wundervald, brunadaviesw@gmail.com.


brunaw/hebart documentation built on June 1, 2022, 8:35 p.m.