predict.utree: Predict method for utree fits.

View source: R/utree.R

predict.utreeR Documentation

Predict method for utree fits.

Description

Obtains predictions from a fitted utree object.

Usage

## S3 method for class 'utree'
predict(object, newdata, type = "uplift", ...)

Arguments

object

A fitted object inheriting from utree.

newdata

A data frame to predict on.

type

The type of predictions required. The default is "uplift" for uplift predictions Alternatively, "node" returns an integer vector of terminal node identifiers.

...

Additional arguments passed to partykit::predict.party.

Value

A numeric vector of predictions.

Author(s)

Leo Guelman leo.guelman@gmail.com

See Also

utree.


leoguelman/uplift2 documentation built on April 15, 2022, 4:34 a.m.