tidylearn (version 0.1.0)

tl_fit_deep: Fit a deep learning model

Description

Fit a deep learning model

Usage

tl_fit_deep(
  data,
  formula,
  is_classification = FALSE,
  hidden_layers = c(32, 16),
  activation = "relu",
  dropout = 0.2,
  epochs = 30,
  batch_size = 32,
  validation_split = 0.2,
  verbose = 0,
  ...
)

Value

A fitted deep learning model

Arguments

data

A data frame containing the training data

formula

A formula specifying the model

is_classification

Logical indicating if this is a classification problem

hidden_layers

Vector of units in each hidden layer (default: c(32, 16))

activation

Activation function for hidden layers (default: "relu")

dropout

Dropout rate for regularization (default: 0.2)

epochs

Number of training epochs (default: 30)

batch_size

Batch size for training (default: 32)

validation_split

Proportion of data for validation (default: 0.2)

verbose

Verbosity mode (0 = silent, 1 = progress bar, 2 = one line per epoch) (default: 0)

...

Additional arguments