org.jpmml.sparkml

Class ModelConverter<T extends org.apache.spark.ml.Model<T> & org.apache.spark.ml.param.shared.HasPredictionCol>

java.lang.Object

org.jpmml.sparkml.TransformerConverter<T>

org.jpmml.sparkml.ModelConverter<T>

Direct Known Subclasses:

AssociationRulesModelConverter, ClusteringModelConverter, PredictionModelConverter

public abstract class ModelConverter<T extends org.apache.spark.ml.Model<T> & org.apache.spark.ml.param.shared.HasPredictionCol>
extends TransformerConverter<T>

Constructor Summary

Constructors

Constructor and Description
ModelConverter(T model)

Method Summary

Modifier and Type	Method and Description
void	checkSchema(org.jpmml.converter.Schema schema)
abstract org.dmg.pmml.Model	encodeModel(org.jpmml.converter.Schema schema)
org.jpmml.converter.Schema	encodeSchema(SparkMLEncoder encoder)
abstract List<org.jpmml.converter.Feature>	getFeatures(SparkMLEncoder encoder)
org.jpmml.converter.Label	getLabel(SparkMLEncoder encoder)
abstract org.dmg.pmml.MiningFunction	getMiningFunction()
T	getModel()
org.dmg.pmml.Model	registerModel(SparkMLEncoder encoder)
List<org.dmg.pmml.OutputField>	registerOutputFields(org.jpmml.converter.Label label, org.dmg.pmml.Model model, SparkMLEncoder encoder)

Methods inherited from class org.jpmml.sparkml.TransformerConverter

getObject, getOption, getOptions, setOptions

Methods inherited from class java.lang.Object

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

Constructor Detail

ModelConverter

public ModelConverter(T model)

Method Detail

getMiningFunction

public abstract org.dmg.pmml.MiningFunction getMiningFunction()

getFeatures

public abstract List<org.jpmml.converter.Feature> getFeatures(SparkMLEncoder encoder)

encodeModel

public abstract org.dmg.pmml.Model encodeModel(org.jpmml.converter.Schema schema)

encodeSchema

public org.jpmml.converter.Schema encodeSchema(SparkMLEncoder encoder)

getLabel

public org.jpmml.converter.Label getLabel(SparkMLEncoder encoder)

checkSchema

public void checkSchema(org.jpmml.converter.Schema schema)

registerOutputFields

public List<org.dmg.pmml.OutputField> registerOutputFields(org.jpmml.converter.Label label,
                                                           org.dmg.pmml.Model model,
                                                           SparkMLEncoder encoder)

registerModel

public org.dmg.pmml.Model registerModel(SparkMLEncoder encoder)

getModel

public T getModel()