org.apache.lens.ml.algo.spark.nb

Class NaiveBayesClassificationModel

java.lang.Object

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

org.apache.lens.ml.algo.lib.ClassifierBaseModel

org.apache.lens.ml.algo.spark.BaseSparkClassificationModel<org.apache.spark.mllib.classification.NaiveBayesModel>

org.apache.lens.ml.algo.spark.nb.NaiveBayesClassificationModel

All Implemented Interfaces:

Serializable

public class NaiveBayesClassificationModel
extends BaseSparkClassificationModel<org.apache.spark.mllib.classification.NaiveBayesModel>

The Class NaiveBayesClassificationModel.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
NaiveBayesClassificationModel(String modelId, org.apache.spark.mllib.classification.NaiveBayesModel model) Instantiates a new naive bayes classification model.

Method Summary

Methods inherited from class org.apache.lens.ml.algo.spark.BaseSparkClassificationModel

getId, predict

Methods inherited from class org.apache.lens.ml.algo.lib.ClassifierBaseModel

getFeatureVector

Methods inherited from class java.lang.Object

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

Constructor Detail

NaiveBayesClassificationModel

public NaiveBayesClassificationModel(String modelId,
                             org.apache.spark.mllib.classification.NaiveBayesModel model)

Instantiates a new naive bayes classification model.

Parameters:

modelId - the model id

model - the model