org.jpmml.sparkml

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

java.lang.Object

org.jpmml.sparkml.TransformerConverter<T>

org.jpmml.sparkml.ModelConverter<T>

Direct Known Subclasses:

ClassificationModelConverter, ClusteringModelConverter, RegressionModelConverter

public abstract class ModelConverter<T extends org.apache.spark.ml.Model<T> & org.apache.spark.ml.param.shared.HasFeaturesCol & 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
abstract org.dmg.pmml.Model	encodeModel(org.jpmml.converter.Schema schema)
org.jpmml.converter.Schema	encodeSchema(SparkMLEncoder encoder)
List<org.jpmml.converter.Feature>	getFeatures(SparkMLEncoder encoder)
org.jpmml.converter.Label	getLabel(SparkMLEncoder encoder)
abstract org.dmg.pmml.MiningFunction	getMiningFunction()
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

getOption, getOptions, getTransformer, 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()

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)

getFeatures

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

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)