dec_tree.fit: Bayesian Decision Tree Fit

Description Usage Arguments Value Author(s)

Description

Fit a Bayesian Decision Tree with a stats-like formula frontend interface.

Usage

1
2
3
dec_tree.fit(formuler, data = NULL, d = NULL, alpha = NULL,
  method = "classification", depth.max = 5L, size = 1L,
  debug = FALSE, ...)

Arguments

formuler

ravioli ravioli give me the formuoli.

data

the data associated with the formuler. Note: if you want an intercept, you must add it ahead of time.

d

the number of features to subsample at each node. Defaults to NULL, which tries every feature.

alpha

the prior parameters for the feature probabilities. A [p] vector. If NULL, samples uniformly. Defaults to NULL.

method

whether you want "classification" or "regression". Defaults to "classification".

depth.max

the maximum allowed tree depth. Defaults to 5L.

size

the minimum allowed number of samples for an individual node. Defaults to 1L.

debug

whether to save the predictors and responses that are categorized. Defaults to FALSE.

...

trailing arguments.

Value

an object of class dec.tree.class containing the following:

tree

the decision tree.

X

The training predictors.

Y

the training responses.

d

d the number of features subsampled at each node.

alpha

the sampling distribution for the features. A [p] vector.

depth.max

the maximum allowed tree depth.

size

the maximum allowed tree depth.

debug

whether to save the predictors and responses that are categorized.

Author(s)

Eric Bridgeford


ebridge2/badmf documentation built on June 4, 2019, 8:53 a.m.