io.cdap.mmds.plugin

Class MLPredictor

java.lang.Object

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

io.cdap.mmds.plugin.MLPredictor

All Implemented Interfaces:

PipelineConfigurable, Serializable

public class MLPredictor
extends SparkCompute<StructuredRecord,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(PipelineConfigurer pipelineConfigurer)
void	initialize(SparkExecutionPluginContext context)
org.apache.spark.api.java.JavaRDD<StructuredRecord>	transform(SparkExecutionPluginContext sparkExecutionPluginContext, org.apache.spark.api.java.JavaRDD<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(PipelineConfigurer pipelineConfigurer)
                       throws IllegalArgumentException

Specified by:

configurePipeline in interface PipelineConfigurable

Overrides:

configurePipeline in class SparkCompute<StructuredRecord,StructuredRecord>

Throws:

IllegalArgumentException

initialize

public void initialize(SparkExecutionPluginContext context)
                throws Exception

Overrides:

initialize in class SparkCompute<StructuredRecord,StructuredRecord>

Throws:

Exception

transform

public org.apache.spark.api.java.JavaRDD<StructuredRecord> transform(SparkExecutionPluginContext sparkExecutionPluginContext,
                                                                     org.apache.spark.api.java.JavaRDD<StructuredRecord> javaRDD)
                                                              throws Exception

Specified by:

transform in class SparkCompute<StructuredRecord,StructuredRecord>

Throws:

Exception