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 org.apache.lens.ml.algo.api.MLModel

getAlgoName, getCreatedAt, getFeatureColumns, 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

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