Package io.github.kirstenali.deepj.nn

Class FNN

java.lang.Object

io.github.kirstenali.deepj.nn.FNN

All Implemented Interfaces:

Layer, Trainable

public final class FNN
extends Object
implements Layer

Constructor Summary

Constructors

Constructor	Description
FNN(int inputSize, int[] hiddenSizes, int outputSize, Supplier<ActivationFunction> hiddenActivationFactory, ActivationFunction outputActivation, Random rnd)
FNN(int inputSize, int[] hiddenSizes, int outputSize, Supplier<ActivationFunction> hiddenActivationFactory, ActivationFunction outputActivation, Random rnd, ParameterFactory parameters)
FNN(int inputSize, int[] hiddenSizes, int outputSize, Supplier<ActivationFunction> hiddenActivationFactory, Random rnd)

Method Summary

Modifier and Type	Method	Description
Tensor	forward(Tensor x)
List<Parameter>	parameters()

Methods inherited from class java.lang.Object

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

Methods inherited from interface io.github.kirstenali.deepj.layers.Layer

forward, supportsIncrementalInference

Methods inherited from interface io.github.kirstenali.deepj.training.Trainable

zeroGrad

Constructor Details

FNN

public FNN(int inputSize,
 int[] hiddenSizes,
 int outputSize,
 Supplier<ActivationFunction> hiddenActivationFactory,
 ActivationFunction outputActivation,
 Random rnd)

FNN

public FNN(int inputSize,
 int[] hiddenSizes,
 int outputSize,
 Supplier<ActivationFunction> hiddenActivationFactory,
 ActivationFunction outputActivation,
 Random rnd,
 ParameterFactory parameters)

FNN

public FNN(int inputSize,
 int[] hiddenSizes,
 int outputSize,
 Supplier<ActivationFunction> hiddenActivationFactory,
 Random rnd)

Method Details

forward

public Tensor forward(Tensor x)

Specified by:

forward in interface Layer

parameters

public List<Parameter> parameters()

Specified by:

parameters in interface Trainable