Interface MultilayerPerceptronParams

All Superinterfaces:
ClassifierParams, HasBlockSize, HasFeaturesCol, HasLabelCol, HasMaxIter, HasPredictionCol, HasProbabilityCol, HasRawPredictionCol, HasSeed, HasSolver, HasStepSize, HasThresholds, HasTol, Identifiable, Params, PredictorParams, ProbabilisticClassifierParams, Serializable, scala.Serializable
All Known Implementing Classes:
MultilayerPerceptronClassificationModel, MultilayerPerceptronClassifier

public interface MultilayerPerceptronParams extends ProbabilisticClassifierParams, HasSeed, HasMaxIter, HasTol, HasStepSize, HasSolver, HasBlockSize
Params for Multilayer Perceptron.
  • Method Details

    • getInitialWeights

      Vector getInitialWeights()
    • getLayers

      int[] getLayers()
    • initialWeights

      Param<Vector> initialWeights()
      The initial weights of the model.

    • layers

      IntArrayParam layers()
      Layer sizes including input size and output size.

    • solver

      Param<String> solver()
      The solver algorithm for optimization. Supported options: "gd" (minibatch gradient descent) or "l-bfgs". Default: "l-bfgs"

      Specified by:
      solver in interface HasSolver