splinetree0.2.0 package

Longitudinal Regression Trees and Forests

avSize

Compute the average tree size in a forest

flatten_predictors

Flattens predictor variable data into one row per person

getBasisMat

Get the basis matrix to be used for this spline tree

getNodeData

Retrieve the subset of the data found at a given terminal node

individual_spline

Get spline coefficients for a single person

nodePlot

Plots the trajectories of each terminal node side by side.

plotImp

Create a barplot of relative variable importance scores.

plotNode

Plot the predicted trajectory for a single node

predict_y_training

Predict responses for the training data

predictCoeffs

Predict spline coefficients for a testset using a spline tree

predictCoeffsForest

Predict spline coefficients for a testset using a splineforest.

predictY

Predictions from a spline tree

predictYForest

Predict responses for a testset using a splineforest.

projectedR2

Computes percent of variation in projected response explained by a spl...

projectedR2Forest

Computes a level-based or shape-based evaluation metric for a splinefo...

pruneForest

Prune each tree in forest using a given complexity parameter.

rpartco

Calculates coordinates for tree plot

spaghettiPlot

Create a faceted spaghetti plot of a splinetree model

spline_eval

Custom rpart eval function.

spline_init

Custom rpart init function

spline_split

Custom rpart split function.

splineForest

Build a spline random forest.

splineforest_split

Custom rpart split function for spline random forests

splineTree

Build a splinetree model.

splineTreePlot

Creates a tree plot of a spline tree.

stPlot

Plots a splinetree.

stPrint

Print a spline tree in the style of print.rpart

terminalNodeSummary

Prints a summary of a terminal node in a tree

tree.depth

Given a list of node numbers, returns the depth at which these appear ...

treeSimilarity

Returns a measure of how similar the two trees are.

treeSize

Returns number of terminal nodes in a tree.

treeSummary

Returns the tree frame.

varImpCoeff

Random Forest Variable Importance based on spline coefficients

varImpY

Random Forest Variable Importance based on Y

yR2

Computes percent of variation in response explained by spline tree.

yR2Forest

Computes a level-based evaluation metric for a splineforest that was b...

Builds regression trees and random forests for longitudinal or functional data using a spline projection method. Implements and extends the work of Yu and Lambert (1999) <doi:10.1080/10618600.1999.10474847>. This method allows trees and forests to be built while considering either level and shape or only shape of response trajectories.

  • Maintainer: Anna Neufeld
  • License: MIT + file LICENSE
  • Last published: 2019-07-18