ggmlR (version 0.8.1)

ggmlr_parsnip_fit_regr: 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,
  ...
)

Value

A fitted `ggmlr_parsnip_model` object.

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.