org.jpmml.sparkml

Class PMMLBuilder

java.lang.Object

org.jpmml.sparkml.PMMLBuilder

public class PMMLBuilder
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
static class	PMMLBuilder.Verification

Constructor Summary

Constructors

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

Method Summary

Modifier and Type	Method and Description
org.dmg.pmml.PMML	build()
byte[]	buildByteArray()
File	buildFile(File file)
PMMLBuilder	extendSchema(Set<String> names)
Map<RegexKey,Map<String,Object>>	getOptions()
org.apache.spark.ml.PipelineModel	getPipelineModel()
org.apache.spark.sql.types.StructType	getSchema()
PMMLBuilder.Verification	getVerification()
PMMLBuilder	putOption(org.apache.spark.ml.PipelineStage pipelineStage, String key, Object value)
PMMLBuilder	putOption(String key, Object value)
PMMLBuilder	putOptions(Map<String,?> map)
PMMLBuilder	putOptions(Pattern pattern, Map<String,?> map)
PMMLBuilder	putOptions(org.apache.spark.ml.PipelineStage pipelineStage, Map<String,?> map)
PMMLBuilder	setPipelineModel(org.apache.spark.ml.PipelineModel pipelineModel)
PMMLBuilder	setSchema(org.apache.spark.sql.types.StructType schema)
PMMLBuilder	verify(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> dataset)
PMMLBuilder	verify(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> dataset, double precision, double zeroThreshold)

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)

PMMLBuilder

public PMMLBuilder(org.apache.spark.sql.types.StructType schema,
                   org.apache.spark.ml.PipelineStage pipelineStage)

Method Detail

build

public org.dmg.pmml.PMML build()

buildByteArray

public byte[] buildByteArray()

buildFile

public File buildFile(File file)
               throws IOException

Throws:

IOException

extendSchema

public PMMLBuilder extendSchema(Set<String> names)

putOption

public PMMLBuilder putOption(String key,
                             Object value)

putOptions

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

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)

putOptions

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

verify

public PMMLBuilder verify(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> dataset)

verify

public PMMLBuilder verify(org.apache.spark.sql.Dataset<org.apache.spark.sql.Row> dataset,
                          double precision,
                          double zeroThreshold)

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<RegexKey,Map<String,Object>> getOptions()

getVerification

public PMMLBuilder.Verification getVerification()