org.jpmml.sparkml

Class FeatureMapper

java.lang.Object

org.jpmml.converter.PMMLMapper

org.jpmml.sparkml.FeatureMapper

public class FeatureMapper
extends org.jpmml.converter.PMMLMapper

Constructor Summary

Constructors

Constructor and Description
FeatureMapper(org.apache.spark.sql.types.StructType schema)

Method Summary

Methods

Modifier and Type	Method and Description
void	append(FeatureConverter<?> converter)
void	append(ModelConverter<?> converter)
org.dmg.pmml.DataField	createDataField(org.dmg.pmml.FieldName name)
org.jpmml.converter.Schema	createSchema(org.apache.spark.ml.Model<?> model)
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)
boolean	hasFeatures(String column)
void	putFeatures(String column, List<org.jpmml.converter.Feature> features)
void	removeDataField(org.dmg.pmml.FieldName name)
org.dmg.pmml.DataField	toCategorical(org.dmg.pmml.FieldName name, List<String> categories)

Methods inherited from class org.jpmml.converter.PMMLMapper

addDataField, addDecorator, addDefineFunction, addDerivedField, createDataField, createDerivedField, encodeHeader, encodePMML, getDataField, getDataFields, getDecorators, getDefineFunction, getDerivedField, getDerivedFields, getField

Methods inherited from class java.lang.Object

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

Constructor Detail

FeatureMapper

public FeatureMapper(org.apache.spark.sql.types.StructType schema)

Method Detail

append

public void append(FeatureConverter<?> converter)

append

public void append(ModelConverter<?> converter)

createSchema

public org.jpmml.converter.Schema createSchema(org.apache.spark.ml.Model<?> model)

toCategorical

public org.dmg.pmml.DataField toCategorical(org.dmg.pmml.FieldName name,
                                   List<String> categories)

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)

putFeatures

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

createDataField

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

removeDataField

public void removeDataField(org.dmg.pmml.FieldName name)