ggmlr_parsnip_fit_regr: parsnip ggml engine: regression fit

In ggmlR: 'GGML' Tensor Operations for Machine Learning

parsnip ggml engine: regression fit

Description

Internal fit wrapper called by parsnip when 'mlp()' is used with 'engine = "ggml"' in regression mode. Not intended for direct use.

Usage

ggmlr_parsnip_fit_regr(
  x,
  y,
  hidden_layers = c(128L, 64L),
  epochs = 10L,
  dropout = 0.2,
  activation = "relu",
  learn_rate = NULL,
  batch_size = 32L,
  verbose = 0L,
  validation_split = 0,
  callbacks = list(),
  optimizer = "adam",
  backend = "auto",
  seed = NULL,
  ...
)

Arguments

x	Numeric matrix of predictors.
y	Numeric response vector.
hidden_layers	Integer vector of hidden layer widths.
epochs	Number of training epochs.
dropout	Dropout rate in '[0, 1)'.
activation	Hidden activation, e.g. '"relu"'.
learn_rate	Optional learning rate (applied via callback).
batch_size	Minibatch size.
verbose	Verbosity level (0/1/2).
validation_split	Fraction in '[0, 1)' for validation.
callbacks	List of ggmlR callbacks.
optimizer	One of '"adam"', '"sgd"'.
backend	One of '"auto"', '"cpu"', '"vulkan"'.
seed	Optional integer RNG seed for reproducible weight init / training.
...	Unused.

Value

A fitted 'ggmlr_parsnip_model' object.

ggmlR documentation built on July 14, 2026, 1:08 a.m.