predictY {splinetree} | R Documentation |
Predictions from a spline tree
Description
Returns a vector of predicted responses for the testData. If testData is ommitted, returns predictions for the training data. This function is most meaningful if model$intercept==TRUE.
Usage
predictY(model, testData = NULL)
Arguments
model |
A model created with splineTree() |
testData |
The data to return predictions for. If ommitted, uses the training data. |
Value
A vector of predictions with rows corresponding to the testdata.
Examples
split_formula <- ~HISP + WHITE + BLACK + SEX + Num_sibs + HGC_FATHER + HGC_MOTHER
tree <- splineTree(split_formula, BMI~AGE, idvar = "ID",
data = nlsySample, degree = 1, df = 3,
intercept = TRUE, cp = 0.005)
plot(predictY(tree), tree$parms$data[[tree$parms$yvar]])
[Package splinetree version 0.2.0 Index]