org.apache.spark.ml.classification

Class LogisticRegression

Object

org.apache.spark.ml.PipelineStage

org.apache.spark.ml.Estimator<M>

org.apache.spark.ml.impl.estimator.Predictor<FeaturesType,E,M>

org.apache.spark.ml.classification.Classifier<FeaturesType,E,M>

org.apache.spark.ml.classification.ProbabilisticClassifier<Vector,LogisticRegression,LogisticRegressionModel>

org.apache.spark.ml.classification.LogisticRegression

All Implemented Interfaces:

java.io.Serializable, Logging, ClassifierParams, LogisticRegressionParams, ProbabilisticClassifierParams, Identifiable, PredictorParams, HasFeaturesCol, HasLabelCol, HasMaxIter, HasPredictionCol, HasProbabilityCol, HasRawPredictionCol, HasRegParam, HasThreshold, Params

public class LogisticRegression
extends ProbabilisticClassifier<Vector,LogisticRegression,LogisticRegressionModel>
implements LogisticRegressionParams

:: AlphaComponent ::

Logistic regression. Currently, this class only supports binary classification.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
LogisticRegression()

Method Summary

Methods

Modifier and Type	Method and Description
LogisticRegression	setMaxIter(int value)
LogisticRegression	setRegParam(double value)
LogisticRegression	setThreshold(double value)

Methods inherited from class org.apache.spark.ml.classification.ProbabilisticClassifier

setProbabilityCol

Methods inherited from class org.apache.spark.ml.classification.Classifier

setRawPredictionCol

Methods inherited from class org.apache.spark.ml.impl.estimator.Predictor

fit, setFeaturesCol, setLabelCol, setPredictionCol, transformSchema

Methods inherited from class org.apache.spark.ml.Estimator

fit, fit, fit

Methods inherited from class Object

equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

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

validateAndTransformSchema

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

getLabelCol, labelCol

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

featuresCol, getFeaturesCol

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

getPredictionCol, predictionCol

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

addOutputColumn, checkInputColumn, explainParams, get, getParam, isSet, paramMap, params, set, set, validate, validate

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

uid

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

getRawPredictionCol, rawPredictionCol

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

getProbabilityCol, probabilityCol

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

getRegParam, regParam

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

getMaxIter, maxIter

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

getThreshold, threshold

Methods inherited from interface org.apache.spark.Logging

initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning

Constructor Detail

LogisticRegression

public LogisticRegression()

Method Detail

setRegParam

public LogisticRegression setRegParam(double value)

setMaxIter

public LogisticRegression setMaxIter(int value)

setThreshold

public LogisticRegression setThreshold(double value)