grapes-greater-than-grapes: bartMachine shrinkage function

In davidaarmstrong/mldiag: Machine Learning Model Diagnostics

Description

bartMachine shrinkage function for the ml_diag function.

Usage

bm_shrink(obj, train, test, ...)

Arguments

obj	An object of class lm or glm with family=binomial.
train	A vector of observations to be used in the training set.
test	A vector of observations to be used in the testing set.
...	Other arguments to be passed down to the bartMachine function.

Value

A list which contains the predicted values for the training set, predicted values for the testing set and the bartMachine model used to generate the predictions.

davidaarmstrong/mldiag documentation built on April 17, 2020, 12:04 a.m.