predict.xrf {xrf} | R Documentation |
Draw predictions from a RuleFit xrf model
## S3 method for class 'xrf'
predict(
object,
newdata,
sparse = TRUE,
lambda = "lambda.min",
type = "response",
...
)
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 |
m <- xrf(Petal.Length ~ ., iris,
xgb_control = list(nrounds = 2, max_depth = 2),
family = 'gaussian')
predictions <- predict(m, iris)