com.kylecorry.lann

Class NeuralNetwork.Builder

java.lang.Object

com.kylecorry.lann.NeuralNetwork.Builder

Enclosing class:

NeuralNetwork

public static class NeuralNetwork.Builder
extends java.lang.Object

Builder for creating neural network instances.

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
NeuralNetwork.Builder	addLayer(int input, int output, Activation function) Adds a layer to the neural network.
NeuralNetwork	build() Builds a neural network instance.
NeuralNetwork.Builder	setRegularization(double lambda)

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

Builder

public Builder()

Method Detail

addLayer

public NeuralNetwork.Builder addLayer(int input,
                                      int output,
                                      Activation function)

Adds a layer to the neural network.

Parameters:

size - The size of the layer.

function - The activation function of the layer.

setRegularization

public NeuralNetwork.Builder setRegularization(double lambda)

build

public NeuralNetwork build()

Builds a neural network instance.