predict.htt: Predictions from a Fitted htt Object

Description

Compute predictions from htt object.

Usage

# S3 method for htt
predict(object, newdata,
        type = c("response", "prob", "node"),
        ...)

Value

A list of predictions, possibly simplified to a numeric vector, numeric matrix or factor.

If type = "response":

the mean of a numeric response and the predicted class for a categorical response is returned.

If type = "prob":

the matrix of class probabilities is returned for a categorical response.

If type = "node":

an integer vector of terminal node identifiers is returned.

Arguments

object: fitted model object of class htt returned by the HTT function.
newdata: an optional data frame in which to look for variables with which to predict, if omitted, the fitted values are used.
type: a character string denoting the type of predicted value returned. For type = "response", the mean of a numeric response and the predicted class for a categorical response is returned. For type = "prob" the matrix of class probabilities is returned for a categorical response. type = "node" returns an integer vector of terminal node identifiers.
...: additional print arguments.

Details

This function is a method for the generic function predict for class htt. It can be invoked by calling predict for an object of the appropriate class, or directly by calling predict.htt regardless of the class of the object.

Examples

Run this code

irishtt <- HTT(Species ~., data = iris)

## the predicted class
predict(irishtt, type = "response")

## class probabilities
predict(irishtt, type = "prob")

## terminal node identifiers
predict(irishtt, type = "node")