org.jpmml.sparkml

Class PMMLBuilder

java.lang.Object

org.jpmml.sparkml.PMMLBuilder

public class PMMLBuilder
extends Object

Constructor Summary

Constructors

Constructor and Description
PMMLBuilder(org.apache.spark.sql.types.StructType schema, org.apache.spark.ml.PipelineModel pipelineModel)

Method Summary

Modifier and Type	Method and Description
org.dmg.pmml.PMML	build()
byte[]	buildByteArray()
Map<String,Map<String,Object>>	getOptions()
org.apache.spark.ml.PipelineModel	getPipelineModel()
org.apache.spark.sql.types.StructType	getSchema()
PMMLBuilder	putOption(org.apache.spark.ml.PipelineStage pipelineStage, String key, Object value)
PMMLBuilder	putOption(String uid, String key, Object value)
PMMLBuilder	putOptions(org.apache.spark.ml.PipelineStage pipelineStage, Map<String,?> map)
PMMLBuilder	putOptions(String uid, Map<String,?> map)
PMMLBuilder	removeOptions(org.apache.spark.ml.PipelineStage pipelineStage)
PMMLBuilder	removeOptions(String uid)
PMMLBuilder	setPipelineModel(org.apache.spark.ml.PipelineModel pipelineModel)
PMMLBuilder	setSchema(org.apache.spark.sql.types.StructType schema)

Methods inherited from class java.lang.Object

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

Constructor Detail

PMMLBuilder

public PMMLBuilder(org.apache.spark.sql.types.StructType schema,
                   org.apache.spark.ml.PipelineModel pipelineModel)

Method Detail

build

public org.dmg.pmml.PMML build()

buildByteArray

public byte[] buildByteArray()

putOption

public PMMLBuilder putOption(org.apache.spark.ml.PipelineStage pipelineStage,
                             String key,
                             Object value)

putOptions

public PMMLBuilder putOptions(org.apache.spark.ml.PipelineStage pipelineStage,
                              Map<String,?> map)

putOption

public PMMLBuilder putOption(String uid,
                             String key,
                             Object value)

putOptions

public PMMLBuilder putOptions(String uid,
                              Map<String,?> map)

removeOptions

public PMMLBuilder removeOptions(org.apache.spark.ml.PipelineStage pipelineStage)

removeOptions

public PMMLBuilder removeOptions(String uid)

getSchema

public org.apache.spark.sql.types.StructType getSchema()

setSchema

public PMMLBuilder setSchema(org.apache.spark.sql.types.StructType schema)

getPipelineModel

public org.apache.spark.ml.PipelineModel getPipelineModel()

setPipelineModel

public PMMLBuilder setPipelineModel(org.apache.spark.ml.PipelineModel pipelineModel)

getOptions

public Map<String,Map<String,Object>> getOptions()