Package smile.base.mlp

Class OutputLayerBuilder

java.lang.Object

smile.base.mlp.LayerBuilder

smile.base.mlp.OutputLayerBuilder

public class OutputLayerBuilder
extends LayerBuilder

The builder of output layers.

Field Summary

Fields inherited from class smile.base.mlp.LayerBuilder

dropout, neurons

Constructor Summary

Constructors

Constructor	Description
OutputLayerBuilder(int neurons, OutputFunction output, Cost cost)	Constructor.

Method Summary

Modifier and Type	Method	Description
OutputLayer	build(int p)	Builds a layer.
String	toString()

Methods inherited from class smile.base.mlp.LayerBuilder

neurons

Methods inherited from class java.lang.Object

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

Constructor Details

OutputLayerBuilder

public OutputLayerBuilder(int neurons,
 OutputFunction output,
 Cost cost)

Constructor.

Parameters:

neurons - the number of neurons.

output - the output function.

cost - the cost function.

Method Details

toString

public String toString()

Overrides:

toString in class Object

build

public OutputLayer build(int p)

Description copied from class: LayerBuilder

Builds a layer.

Specified by:

build in class LayerBuilder

Parameters:

p - the number of input variables (not including bias value).

Returns:

a layer.