org.apache.lens.ml.spark.models
Class NaiveBayesClassificationModel

java.lang.Object
  org.apache.lens.ml.MLModel<Double>
      org.apache.lens.ml.ClassifierBaseModel
          org.apache.lens.ml.spark.models.BaseSparkClassificationModel<org.apache.spark.mllib.classification.NaiveBayesModel>
              org.apache.lens.ml.spark.models.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

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.spark.models.BaseSparkClassificationModel

getId, predict

Methods inherited from class org.apache.lens.ml.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