aml_neural_network: AML neural network

Description Usage Arguments Value

View source: R/neural_network.R

Description

Trains a simple back-propagation neural network.

Usage

1
2
aml_neural_network(sizes, learning_rate, data, response, epochs,
  verbose = FALSE)

Arguments

sizes

Vector of layer sizes

learning_rate

Learning rate for gradient descent

data

Input data.frame for training network, dimension nxp where p is sizes[1]

response

Response vector of size nx1 corresponding to the training data

epochs

Number of epochs to use in network training

verbose

Logical option for status printing

Value

Trained neural network object of class aml_neural_network


jmwerner/ArtisanalMachineLearning documentation built on Jan. 7, 2020, 1:50 a.m.