predict.xrf {xrf}

Draw predictions from a RuleFit xrf model

Description

Draw predictions from a RuleFit xrf model

Usage

## S3 method for class 'xrf'
predict(
  object,
  newdata,
  sparse = TRUE,
  lambda = "lambda.min",
  type = "response",
  ...
)

Arguments

object

an object of class "xrf"

newdata

data to predict on

sparse

a logical indicating whether a sparse design matrix should be used

lambda

the lasso penalty parameter to be applied

type

the type of predicted value produced

...

ignored arguments

Examples

m <- xrf(Petal.Length ~ ., iris,
         xgb_control = list(nrounds = 2, max_depth = 2),
         family = 'gaussian')
predictions <- predict(m, iris)