org.jpmml.sparkml

Class ClusteringModelConverter<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>

org.jpmml.sparkml.ClusteringModelConverter<T>

Direct Known Subclasses:

KMeansModelConverter

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

Constructor Summary

Constructors

Constructor and Description
ClusteringModelConverter(T model)

Method Summary

Modifier and Type	Method and Description
org.dmg.pmml.MiningFunction	getMiningFunction()
abstract int	getNumberOfClusters()
List<org.dmg.pmml.OutputField>	registerOutputFields(org.jpmml.converter.Label label, SparkMLEncoder encoder)

Methods inherited from class org.jpmml.sparkml.ModelConverter

encodeModel, encodeSchema, registerModel

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

ClusteringModelConverter

public ClusteringModelConverter(T model)

Method Detail

getNumberOfClusters

public abstract int getNumberOfClusters()

getMiningFunction

public org.dmg.pmml.MiningFunction getMiningFunction()

Specified by:

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

registerOutputFields

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

Overrides:

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