wrap_caret: Wrap caret models for mlS3

In mlS3: Unified S3 Interface to Machine Learning Models

Wrap caret models for mlS3

Description

Minimal wrapper around caret::train with no tuning. Hyperparameters can be passed via ... as named arguments.

Usage

wrap_caret(X, y, method = "rf", ...)

Arguments

X	Feature matrix or data frame
y	Response vector
method	caret model method (default "rf")
...	Named hyperparameters (e.g., mtry = 3, ntree = 500)

Value

Object with class "mlS3_caret"

Examples

# Only runs if caret is installed

data(mtcars)

# Prepare data
X_reg <- mtcars[, -1]  # All except mpg
y_reg <- mtcars$mpg     # Target variable

# Split into train/test
set.seed(123)
idx_reg <- sample(nrow(X_reg), 0.7 * nrow(X_reg))
X_train <- X_reg[idx_reg, ]
y_train <- y_reg[idx_reg]
X_test <- X_reg[-idx_reg, ]
y_test <- y_reg[-idx_reg]

mod <- wrap_caret(X_train, y_train, method = "rf", mtry = 3)
(pred <- predict(mod, X_test))

mlS3 documentation built on April 29, 2026, 1:08 a.m.