generates the code to create the neural network model.
nn_model(data, variable.pred, hidden = c(1), threshold = 0.1, stepmax = 2000)
dataframe
the name of the variable to be predicted.
a vector of integers specifying the number of hidden neurons (vertices) in each layer.
a numeric value specifying the threshold for the partial derivatives of the error function as stopping criteria.
the maximum steps for the training of the neural network. Reaching this maximum leads to a stop of the neural network's training process.
neuralnet