mlspatial (version 0.1.1)

train_xgb: Train XGBoost model

Description

Train XGBoost model

Usage

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

Value

A trained xgboost model object.

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.

learning_rate

Learning rate for boosting.

Details

Trains an XGBoost regression model.

Examples

# \donttest{
# 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)
# }