tidylearn (version 0.1.0)

tl_fit_nn: Fit a neural network model

Description

Fit a neural network model

Usage

tl_fit_nn(
  data,
  formula,
  is_classification = FALSE,
  size = 5,
  decay = 0,
  maxit = 100,
  trace = FALSE,
  ...
)

Value

A fitted neural network 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

size

Number of units in the hidden layer (default: 5)

decay

Weight decay parameter (default: 0)

maxit

Maximum number of iterations (default: 100)

trace

Logical; whether to print progress (default: FALSE)

...

Additional arguments to pass to nnet()