Package smile.base.mlp

Class HiddenLayerBuilder

java.lang.Object

smile.base.mlp.LayerBuilder

smile.base.mlp.HiddenLayerBuilder

public class HiddenLayerBuilder
extends LayerBuilder

The builder of hidden layers.

Field Summary

Fields inherited from class smile.base.mlp.LayerBuilder

dropout, neurons

Constructor Summary

Constructors

Constructor	Description
HiddenLayerBuilder(int neurons, double dropout, ActivationFunction activation)	Constructor.

Method Summary

Modifier and Type	Method	Description
HiddenLayer	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

HiddenLayerBuilder

public HiddenLayerBuilder(int neurons,
 double dropout,
 ActivationFunction activation)

Constructor.

Parameters:

neurons - the number of neurons.

dropout - the dropout rate.

activation - the activation function.

Method Details

toString

public String toString()

Overrides:

toString in class Object

build

public HiddenLayer 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.