regressoR (version 2.0.1)

nn_model: nn_model

Description

generates the code to create the neural network model.

Usage

nn_model(data, variable.pred, hidden = c(1), threshold = 0.1, stepmax = 2000)

Arguments

data

dataframe

variable.pred

the name of the variable to be predicted.

hidden

a vector of integers specifying the number of hidden neurons (vertices) in each layer.

threshold

a numeric value specifying the threshold for the partial derivatives of the error function as stopping criteria.

stepmax

the maximum steps for the training of the neural network. Reaching this maximum leads to a stop of the neural network's training process.

See Also

neuralnet