org.apache.spark.ml.classification

Class LogisticRegression

Object

org.apache.spark.ml.PipelineStage

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

org.apache.spark.ml.classification.LogisticRegression

All Implemented Interfaces:

java.io.Serializable, Logging, LogisticRegressionParams, Identifiable, HasFeaturesCol, HasLabelCol, HasMaxIter, HasPredictionCol, HasRegParam, HasScoreCol, HasThreshold, Params

public class LogisticRegression
extends Estimator<LogisticRegressionModel>
implements LogisticRegressionParams

Logistic regression.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
LogisticRegression()

Method Summary

Methods

Modifier and Type	Method and Description
LogisticRegressionModel	fit(SchemaRDD dataset, ParamMap paramMap) Fits a single model to the input data with provided parameter map.
LogisticRegression	setFeaturesCol(String value)
LogisticRegression	setLabelCol(String value)
LogisticRegression	setMaxIter(int value)
LogisticRegression	setPredictionCol(String value)
LogisticRegression	setRegParam(double value)
LogisticRegression	setScoreCol(String value)
LogisticRegression	setThreshold(double value)
org.apache.spark.sql.catalyst.types.StructType	transformSchema(org.apache.spark.sql.catalyst.types.StructType schema, ParamMap paramMap) Derives the output schema from the input schema and parameters.

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

fit, fit, fit, fit, 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.LogisticRegressionParams

validateAndTransformSchema

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.HasLabelCol

getLabelCol, labelCol

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

getThreshold, threshold

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

featuresCol, getFeaturesCol

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

getScoreCol, scoreCol

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

getPredictionCol, predictionCol

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

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

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

uid

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)

setLabelCol

public LogisticRegression setLabelCol(String value)

setThreshold

public LogisticRegression setThreshold(double value)

setFeaturesCol

public LogisticRegression setFeaturesCol(String value)

setScoreCol

public LogisticRegression setScoreCol(String value)

setPredictionCol

public LogisticRegression setPredictionCol(String value)

fit

public LogisticRegressionModel fit(SchemaRDD dataset,
                          ParamMap paramMap)

Description copied from class: Estimator

Fits a single model to the input data with provided parameter map.

Specified by:

fit in class Estimator<LogisticRegressionModel>

Parameters:

dataset - input dataset

paramMap - parameter map

Returns:

fitted model

transformSchema

public org.apache.spark.sql.catalyst.types.StructType transformSchema(org.apache.spark.sql.catalyst.types.StructType schema,
                                                             ParamMap paramMap)

Description copied from class: PipelineStage

Derives the output schema from the input schema and parameters.

Specified by:

transformSchema in class PipelineStage