train_xgb: Train XGBoost model

View source: R/train_MLmodels.R

train_xgbR Documentation

Train XGBoost model

Description

Train XGBoost model

Usage

train_xgb(data, formula, nrounds = 100, max_depth = 4, eta = 0.1)

Arguments

data

A data frame with the training data.

formula

A formula defining the model structure.

nrounds

Number of boosting iterations.

max_depth

Maximum tree depth.

eta

Learning rate.

Details

Trains an XGBoost regression model.

Value

A trained xgboost model object.

Examples


# Load required package
library(xgboost)

# Use built-in dataset
data(mtcars)

# Define regression formula
xgb_formula <- mpg ~ cyl + disp + hp + wt

# Train XGBoost model
xgb_model <- train_xgb(data = mtcars, formula = xgb_formula, nrounds = 50)

# Print model summary
print(xgb_model)


mlspatial documentation built on Aug. 27, 2025, 1:09 a.m.