org.apache.lens.ml.algo.spark.kmeans

Class KMeansClusteringModel

java.lang.Object

org.apache.lens.ml.algo.api.MLModel<Integer>

org.apache.lens.ml.algo.spark.kmeans.KMeansClusteringModel

All Implemented Interfaces:

Serializable

public class KMeansClusteringModel
extends MLModel<Integer>

The Class KMeansClusteringModel.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
KMeansClusteringModel(String modelId, org.apache.spark.mllib.clustering.KMeansModel model) Instantiates a new k means clustering model.

Method Summary

Modifier and Type	Method and Description
Integer	predict(Object... args) Predict.

Methods inherited from class org.apache.lens.ml.algo.api.MLModel

getAlgoName, getCreatedAt, getFeatureColumns, getId, getLabelColumn, getParams, getTable, setAlgoName, setCreatedAt, setFeatureColumns, setId, setLabelColumn, setParams, setTable, toString

Methods inherited from class java.lang.Object

equals, getClass, hashCode, notify, notifyAll, wait, wait, wait

Constructor Detail

KMeansClusteringModel

public KMeansClusteringModel(String modelId,
                             org.apache.spark.mllib.clustering.KMeansModel model)

Instantiates a new k means clustering model.

Parameters:

modelId - the model id

model - the model

Method Detail

predict

public Integer predict(Object... args)

Description copied from class: MLModel

Predict.

Specified by:

predict in class MLModel<Integer>

Parameters:

args - the args

Returns:

the prediction