xrf (version 0.2.0)

predict.xrf: Draw predictions from a RuleFit xrf model

Description

Draw predictions from a RuleFit xrf model

Usage

# S3 method for 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

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

# }

Run the code above in your browser using DataCamp Workspace