org.jpmml.sparkml.model

Class KMeansModelConverter

java.lang.Object

org.jpmml.sparkml.TransformerConverter<T>

org.jpmml.sparkml.ModelConverter<T>

org.jpmml.sparkml.ClusteringModelConverter<org.apache.spark.ml.clustering.KMeansModel>

org.jpmml.sparkml.model.KMeansModelConverter

public class KMeansModelConverter
extends ClusteringModelConverter<org.apache.spark.ml.clustering.KMeansModel>

Constructor Summary

Constructors

Constructor and Description
KMeansModelConverter(org.apache.spark.ml.clustering.KMeansModel model)

Method Summary

Modifier and Type	Method and Description
org.dmg.pmml.clustering.ClusteringModel	encodeModel(org.jpmml.converter.Schema schema)
int	getNumberOfClusters()

Methods inherited from class org.jpmml.sparkml.ClusteringModelConverter

getFeatures, getMiningFunction, registerOutputFields

Methods inherited from class org.jpmml.sparkml.ModelConverter

checkSchema, encodeSchema, getLabel, getModel, registerModel

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

KMeansModelConverter

public KMeansModelConverter(org.apache.spark.ml.clustering.KMeansModel model)

Method Detail

getNumberOfClusters

public int getNumberOfClusters()

Specified by:

getNumberOfClusters in class ClusteringModelConverter<org.apache.spark.ml.clustering.KMeansModel>

encodeModel

public org.dmg.pmml.clustering.ClusteringModel encodeModel(org.jpmml.converter.Schema schema)

Specified by:

encodeModel in class ModelConverter<org.apache.spark.ml.clustering.KMeansModel>