train_nn_impl_dataset: train_nn implementation for torch datasets

In kindling: Higher-Level Interface of 'torch' Package to Auto-Train Neural Networks

train_nn implementation for torch datasets

Description

train_nn implementation for torch datasets

Usage

train_nn_impl_dataset(
  dataset,
  no_x,
  no_y,
  is_classification,
  hidden_neurons,
  activations = NULL,
  output_activation = NULL,
  bias = TRUE,
  epochs = 100,
  batch_size = 32,
  penalty = 0,
  mixture = 0,
  learn_rate = 0.001,
  optimizer = "adam",
  optimizer_args = list(),
  loss = "mse",
  validation_split = 0,
  device = NULL,
  verbose = FALSE,
  cache_weights = FALSE,
  flatten_input = TRUE,
  arch = NULL,
  fit_class = "nn_fit_ds"
)

kindling documentation built on March 3, 2026, 9:07 a.m.