public class LogisticRegression extends Estimator<LogisticRegressionModel> implements LogisticRegressionParams
Constructor and Description |
---|
LogisticRegression() |
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.
|
equals, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
validateAndTransformSchema
getRegParam, regParam
getMaxIter, maxIter
getLabelCol, labelCol
getThreshold, threshold
featuresCol, getFeaturesCol
getScoreCol, scoreCol
getPredictionCol, predictionCol
explainParams, get, getParam, isSet, paramMap, params, set, validate, validate
uid
initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning
public LogisticRegression setRegParam(double value)
public LogisticRegression setMaxIter(int value)
public LogisticRegression setLabelCol(String value)
public LogisticRegression setThreshold(double value)
public LogisticRegression setFeaturesCol(String value)
public LogisticRegression setScoreCol(String value)
public LogisticRegression setPredictionCol(String value)
public LogisticRegressionModel fit(SchemaRDD dataset, ParamMap paramMap)
Estimator
fit
in class Estimator<LogisticRegressionModel>
dataset
- input datasetparamMap
- parameter mappublic org.apache.spark.sql.catalyst.types.StructType transformSchema(org.apache.spark.sql.catalyst.types.StructType schema, ParamMap paramMap)
PipelineStage
transformSchema
in class PipelineStage