xgb_shrink: Xgboost shrinkage function

In davidaarmstrong/mldiag: Machine Learning Model Diagnostics

Description

Xgboost shrinkage function for the ml_diag function.

Usage

xgb_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 xgboost function.

Value

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

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