org.jpmml.sparkml

Class SparkMLEncoder

java.lang.Object

org.jpmml.converter.PMMLEncoder

org.jpmml.converter.ModelEncoder

org.jpmml.sparkml.SparkMLEncoder

public class SparkMLEncoder
extends org.jpmml.converter.ModelEncoder

Field Summary

Fields inherited from class org.jpmml.converter.PMMLEncoder

VERSION

Constructor Summary

Constructors

Constructor and Description
SparkMLEncoder(org.apache.spark.sql.types.StructType schema, ConverterFactory converterFactory)

Method Summary

Modifier and Type	Method and Description
org.dmg.pmml.DataField	createDataField(org.dmg.pmml.FieldName name)
org.dmg.pmml.PMML	encodePMML(org.dmg.pmml.Model model)
ConverterFactory	getConverterFactory()
List<org.jpmml.converter.Feature>	getFeatures(String column)
List<org.jpmml.converter.Feature>	getFeatures(String column, int[] indices)
org.jpmml.converter.Feature	getOnlyFeature(String column)
org.apache.spark.sql.types.StructType	getSchema()
boolean	hasFeatures(String column)
void	putFeatures(String column, List<org.jpmml.converter.Feature> features)
void	putOnlyFeature(String column, org.jpmml.converter.Feature feature)

Methods inherited from class org.jpmml.converter.ModelEncoder

addDecorator, addDecorator, addFeatureImportance, addFeatureImportance, addTransformer, getDecorators, getFeatureImportances, getTransformers, getUnivariateStats, putUnivariateStats, putUnivariateStats, transferFeatureImportances, transferFeatureImportances

Methods inherited from class org.jpmml.converter.PMMLEncoder

addDataField, addDefineFunction, addDerivedField, createDataField, createDataField, createDerivedField, createDerivedField, encodeHeader, encodePMML, ensureDerivedField, getDataField, getDataFields, getDefineFunction, getDefineFunctions, getDerivedField, getDerivedFields, getField, removeDataField, removeDerivedField, toCategorical, toContinuous, toOrdinal

Methods inherited from class java.lang.Object

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

Constructor Detail

SparkMLEncoder

public SparkMLEncoder(org.apache.spark.sql.types.StructType schema,
                      ConverterFactory converterFactory)

Method Detail

encodePMML

public org.dmg.pmml.PMML encodePMML(org.dmg.pmml.Model model)

Overrides:

encodePMML in class org.jpmml.converter.ModelEncoder

hasFeatures

public boolean hasFeatures(String column)

getOnlyFeature

public org.jpmml.converter.Feature getOnlyFeature(String column)

getFeatures

public List<org.jpmml.converter.Feature> getFeatures(String column)

getFeatures

public List<org.jpmml.converter.Feature> getFeatures(String column,
                                                     int[] indices)

putOnlyFeature

public void putOnlyFeature(String column,
                           org.jpmml.converter.Feature feature)

putFeatures

public void putFeatures(String column,
                        List<org.jpmml.converter.Feature> features)

createDataField

public org.dmg.pmml.DataField createDataField(org.dmg.pmml.FieldName name)

getSchema

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

getConverterFactory

public ConverterFactory getConverterFactory()