org.deeplearning4j.nn.layers

Class OutputLayer.Builder

java.lang.Object

org.deeplearning4j.nn.layers.OutputLayer.Builder

Enclosing class:

OutputLayer

public static class OutputLayer.Builder
extends Object

Constructor Summary

Constructors

Constructor and Description
Builder()

Method Summary

Modifier and Type	Method and Description
OutputLayer	build()
OutputLayer.Builder	configure(NeuralNetConfiguration conf)
OutputLayer.Builder	input(org.nd4j.linalg.api.ndarray.INDArray input)
OutputLayer.Builder	withBias(org.nd4j.linalg.api.ndarray.INDArray b)
OutputLayer.Builder	withLabels(org.nd4j.linalg.api.ndarray.INDArray labels)
OutputLayer.Builder	withWeights(org.nd4j.linalg.api.ndarray.INDArray W)

Methods inherited from class java.lang.Object

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

Constructor Detail

Builder

public Builder()

Method Detail

configure

public OutputLayer.Builder configure(NeuralNetConfiguration conf)

input

public OutputLayer.Builder input(org.nd4j.linalg.api.ndarray.INDArray input)

withLabels

public OutputLayer.Builder withLabels(org.nd4j.linalg.api.ndarray.INDArray labels)

withWeights

public OutputLayer.Builder withWeights(org.nd4j.linalg.api.ndarray.INDArray W)

withBias

public OutputLayer.Builder withBias(org.nd4j.linalg.api.ndarray.INDArray b)

build

public OutputLayer build()