org.apache.spark.ml.classification

Class LogisticRegressionModel

Object

org.apache.spark.ml.PipelineStage

org.apache.spark.ml.Transformer

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

org.apache.spark.ml.classification.LogisticRegressionModel

All Implemented Interfaces:

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

public class LogisticRegressionModel
extends Model<LogisticRegressionModel>
implements LogisticRegressionParams

:: AlphaComponent :: Model produced by LogisticRegression.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
LogisticRegressionModel(LogisticRegression parent, ParamMap fittingParamMap, Vector weights)

Method Summary

Methods

Modifier and Type	Method and Description
ParamMap	fittingParamMap() Fitting parameters, such that parent.fit(..., fittingParamMap) could reproduce the model.
LogisticRegression	parent() The parent estimator that produced this model.
LogisticRegressionModel	setFeaturesCol(String value)
LogisticRegressionModel	setPredictionCol(String value)
LogisticRegressionModel	setScoreCol(String value)
LogisticRegressionModel	setThreshold(double value)
SchemaRDD	transform(SchemaRDD dataset, ParamMap paramMap) Transforms the dataset with provided parameter map as additional parameters.
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.Transformer

transform, transform, transform, transform, transform

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

LogisticRegressionModel

public LogisticRegressionModel(LogisticRegression parent,
                       ParamMap fittingParamMap,
                       Vector weights)

Method Detail

parent

public LogisticRegression parent()

Description copied from class: Model

The parent estimator that produced this model.

Specified by:

parent in class Model<LogisticRegressionModel>

fittingParamMap

public ParamMap fittingParamMap()

Description copied from class: Model

Fitting parameters, such that parent.fit(..., fittingParamMap) could reproduce the model.

Specified by:

fittingParamMap in class Model<LogisticRegressionModel>

setThreshold

public LogisticRegressionModel setThreshold(double value)

setFeaturesCol

public LogisticRegressionModel setFeaturesCol(String value)

setScoreCol

public LogisticRegressionModel setScoreCol(String value)

setPredictionCol

public LogisticRegressionModel setPredictionCol(String value)

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

transform

public SchemaRDD transform(SchemaRDD dataset,
                  ParamMap paramMap)

Description copied from class: Transformer

Transforms the dataset with provided parameter map as additional parameters.

Specified by:

transform in class Transformer

Parameters:

dataset - input dataset

paramMap - additional parameters, overwrite embedded params

Returns:

transformed dataset