io.cdap.mmds.plugin

Class MLPredictor

java.lang.Object

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

io.cdap.mmds.plugin.MLPredictor

All Implemented Interfaces:

io.cdap.cdap.etl.api.PipelineConfigurable, Serializable

public class MLPredictor
extends io.cdap.cdap.etl.api.batch.SparkCompute<io.cdap.cdap.api.data.format.StructuredRecord,io.cdap.cdap.api.data.format.StructuredRecord>

Uses a deployed model to add a prediction field to incoming records.

See Also:

Serialized Form

Field Summary

Fields inherited from class io.cdap.cdap.etl.api.batch.SparkCompute

PLUGIN_TYPE

Constructor Summary

Constructors

Constructor and Description
MLPredictor(PredictorConf conf)

Method Summary

Modifier and Type	Method and Description
void	configurePipeline(io.cdap.cdap.etl.api.PipelineConfigurer pipelineConfigurer)
void	initialize(io.cdap.cdap.etl.api.batch.SparkExecutionPluginContext context)
org.apache.spark.api.java.JavaRDD<io.cdap.cdap.api.data.format.StructuredRecord>	transform(io.cdap.cdap.etl.api.batch.SparkExecutionPluginContext sparkExecutionPluginContext, org.apache.spark.api.java.JavaRDD<io.cdap.cdap.api.data.format.StructuredRecord> javaRDD)

Methods inherited from class java.lang.Object

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

Constructor Detail

MLPredictor

public MLPredictor(PredictorConf conf)

Method Detail

configurePipeline

public void configurePipeline(io.cdap.cdap.etl.api.PipelineConfigurer pipelineConfigurer)
                       throws IllegalArgumentException

Specified by:

configurePipeline in interface io.cdap.cdap.etl.api.PipelineConfigurable

Overrides:

configurePipeline in class io.cdap.cdap.etl.api.batch.SparkCompute<io.cdap.cdap.api.data.format.StructuredRecord,io.cdap.cdap.api.data.format.StructuredRecord>

Throws:

IllegalArgumentException

initialize

public void initialize(io.cdap.cdap.etl.api.batch.SparkExecutionPluginContext context)
                throws Exception

Overrides:

initialize in class io.cdap.cdap.etl.api.batch.SparkCompute<io.cdap.cdap.api.data.format.StructuredRecord,io.cdap.cdap.api.data.format.StructuredRecord>

Throws:

Exception

transform

public org.apache.spark.api.java.JavaRDD<io.cdap.cdap.api.data.format.StructuredRecord> transform(io.cdap.cdap.etl.api.batch.SparkExecutionPluginContext sparkExecutionPluginContext,
                                                                                                  org.apache.spark.api.java.JavaRDD<io.cdap.cdap.api.data.format.StructuredRecord> javaRDD)
                                                                                           throws Exception

Specified by:

transform in class io.cdap.cdap.etl.api.batch.SparkCompute<io.cdap.cdap.api.data.format.StructuredRecord,io.cdap.cdap.api.data.format.StructuredRecord>

Throws:

Exception