co.cask.hydrator.plugin.spark

Class LogisticRegressionClassifier

java.lang.Object

co.cask.cdap.etl.api.batch.SparkCompute<co.cask.cdap.api.data.format.StructuredRecord,co.cask.cdap.api.data.format.StructuredRecord>

co.cask.hydrator.plugin.spark.SparkMLPredictor

co.cask.hydrator.plugin.spark.LogisticRegressionClassifier

All Implemented Interfaces:

co.cask.cdap.etl.api.PipelineConfigurable, Serializable

public class LogisticRegressionClassifier
extends SparkMLPredictor

SparkCompute that uses a trained Logistic Regression model to classify and tag input records.

See Also:

Serialized Form

Nested Class Summary

Nested classes/interfaces inherited from class co.cask.hydrator.plugin.spark.SparkMLPredictor

SparkMLPredictor.MLPredictorConfig

Field Summary

Fields

Modifier and Type	Field and Description
static String	PLUGIN_NAME

Fields inherited from class co.cask.cdap.etl.api.batch.SparkCompute

PLUGIN_TYPE

Constructor Summary

Constructors

Constructor and Description
LogisticRegressionClassifier(SparkMLPredictor.MLPredictorConfig config)

Method Summary

Methods

Modifier and Type	Method and Description
void	initialize(org.apache.spark.SparkContext context, String modelPath)
double	predict(org.apache.spark.mllib.linalg.Vector vector)

Methods inherited from class co.cask.hydrator.plugin.spark.SparkMLPredictor

configurePipeline, initialize, transform

Methods inherited from class java.lang.Object

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

Field Detail

PLUGIN_NAME

public static final String PLUGIN_NAME

See Also:

Constant Field Values

Constructor Detail

LogisticRegressionClassifier

public LogisticRegressionClassifier(SparkMLPredictor.MLPredictorConfig config)

Method Detail

initialize

public void initialize(org.apache.spark.SparkContext context,
              String modelPath)

Specified by:

initialize in class SparkMLPredictor

predict

public double predict(org.apache.spark.mllib.linalg.Vector vector)

Specified by:

predict in class SparkMLPredictor