tl_predict_xgboost: Predict using an XGBoost model

View source: R/supervised-xgboost.R

tl_predict_xgboostR Documentation

Predict using an XGBoost model

Description

Predict using an XGBoost model

Usage

tl_predict_xgboost(model, new_data, type = "response", ntreelimit = NULL, ...)

Arguments

model

A tidylearn XGBoost model object

new_data

A data frame containing the new data

type

Type of prediction: "response" (default), "prob" (for classification), "class" (for classification)

ntreelimit

Limit number of trees used for prediction (default: NULL, uses all trees)

...

Additional arguments

Value

Predictions


tidylearn documentation built on Feb. 6, 2026, 5:07 p.m.