Class OutputLayerBuilder

java.lang.Object
smile.base.mlp.LayerBuilder
smile.base.mlp.OutputLayerBuilder

public class OutputLayerBuilder extends LayerBuilder
The builder of output layers.
  • 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.