org.apache.spark.ml.classification

Interface MultilayerPerceptronParams

All Superinterfaces:

ClassifierParams, HasFeaturesCol, HasLabelCol, HasMaxIter, HasPredictionCol, HasProbabilityCol, HasRawPredictionCol, HasSeed, HasSolver, HasStepSize, HasThresholds, HasTol, Identifiable, Params, PredictorParams, ProbabilisticClassifierParams, java.io.Serializable

All Known Implementing Classes:

MultilayerPerceptronClassifier

public interface MultilayerPerceptronParams
extends ProbabilisticClassifierParams, HasSeed, HasMaxIter, HasTol, HasStepSize, HasSolver

Params for Multilayer Perceptron.

Method Summary

Modifier and Type	Method and Description
IntParam	blockSize() Block size for stacking input data in matrices to speed up the computation.
int	getBlockSize()
Vector	getInitialWeights()
int[]	getLayers()
Param<Vector>	initialWeights() The initial weights of the model.
IntArrayParam	layers() Layer sizes including input size and output size.
Param<String>	solver() The solver algorithm for optimization.

Methods inherited from interface org.apache.spark.ml.classification.ProbabilisticClassifierParams

validateAndTransformSchema

Methods inherited from interface org.apache.spark.ml.param.shared.HasLabelCol

getLabelCol, labelCol

Methods inherited from interface org.apache.spark.ml.param.shared.HasFeaturesCol

featuresCol, getFeaturesCol

Methods inherited from interface org.apache.spark.ml.param.shared.HasPredictionCol

getPredictionCol, predictionCol

Methods inherited from interface org.apache.spark.ml.param.Params

clear, copy, copyValues, defaultCopy, defaultParamMap, explainParam, explainParams, extractParamMap, extractParamMap, get, getDefault, getOrDefault, getParam, hasDefault, hasParam, isDefined, isSet, paramMap, params, set, set, set, setDefault, setDefault, shouldOwn

Methods inherited from interface org.apache.spark.ml.util.Identifiable

toString, uid

Methods inherited from interface org.apache.spark.ml.param.shared.HasRawPredictionCol

getRawPredictionCol, rawPredictionCol

Methods inherited from interface org.apache.spark.ml.param.shared.HasProbabilityCol

getProbabilityCol, probabilityCol

Methods inherited from interface org.apache.spark.ml.param.shared.HasThresholds

getThresholds, thresholds

Methods inherited from interface org.apache.spark.ml.param.shared.HasSeed

getSeed, seed

Methods inherited from interface org.apache.spark.ml.param.shared.HasMaxIter

getMaxIter, maxIter

Methods inherited from interface org.apache.spark.ml.param.shared.HasTol

getTol, tol

Methods inherited from interface org.apache.spark.ml.param.shared.HasStepSize

getStepSize, stepSize

Methods inherited from interface org.apache.spark.ml.param.shared.HasSolver

getSolver

Method Detail

blockSize

IntParam blockSize()

Block size for stacking input data in matrices to speed up the computation. Data is stacked within partitions. If block size is more than remaining data in a partition then it is adjusted to the size of this data. Recommended size is between 10 and 1000. Default: 128

Returns:

(undocumented)

getBlockSize

int getBlockSize()

getInitialWeights

Vector getInitialWeights()

getLayers

int[] getLayers()

initialWeights

Param<Vector> initialWeights()

The initial weights of the model.

Returns:

(undocumented)

layers

IntArrayParam layers()

Layer sizes including input size and output size.

Returns:

(undocumented)

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

Returns:

(undocumented)