bart-internal: Developer functions for predictions via BART models

bart-internalR Documentation

Developer functions for predictions via BART models

Description

Developer functions for predictions via BART models

Usage

bartMachine_interval_calc(new_data, obj, ci = TRUE, level = 0.95)

dbart_predict_calc(obj, new_data, type, level = 0.95, std_err = FALSE)

Arguments

new_data

A rectangular data object, such as a data frame.

obj

A parsnip object.

ci

Confidence (TRUE) or prediction interval (FALSE)

level

Confidence level.

type

A single character value or NULL. Possible values are "numeric", "class", "prob", "conf_int", "pred_int", "quantile", "time", "hazard", "survival", or "raw". When NULL, predict() will choose an appropriate value based on the model's mode.

std_err

Attach column for standard error of prediction or not.


parsnip documentation built on June 24, 2024, 5:14 p.m.