org.apache.spark.mllib.classification

Class NaiveBayesModel

Object

org.apache.spark.mllib.classification.NaiveBayesModel

All Implemented Interfaces:

java.io.Serializable, ClassificationModel

public class NaiveBayesModel
extends Object
implements ClassificationModel, scala.Serializable

Model for Naive Bayes Classifiers.

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
NaiveBayesModel(double[] labels, double[] pi, double[][] theta)

Method Summary

Methods

Modifier and Type	Method and Description
double[]	labels()
double[]	pi()
RDD<Object>	predict(RDD<Vector> testData) Predict values for the given data set using the model trained.
double	predict(Vector testData) Predict values for a single data point using the model trained.
double[][]	theta()

Methods inherited from class Object

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

Methods inherited from interface org.apache.spark.mllib.classification.ClassificationModel

predict

Constructor Detail

NaiveBayesModel

public NaiveBayesModel(double[] labels,
               double[] pi,
               double[][] theta)

Method Detail

labels

public double[] labels()

pi

public double[] pi()

theta

public double[][] theta()

predict

public RDD<Object> predict(RDD<Vector> testData)

Description copied from interface: ClassificationModel

Predict values for the given data set using the model trained.

Specified by:

predict in interface ClassificationModel

Parameters:

testData - RDD representing data points to be predicted

Returns:

an RDD[Double] where each entry contains the corresponding prediction

predict

public double predict(Vector testData)

Description copied from interface: ClassificationModel

Predict values for a single data point using the model trained.

Specified by:

predict in interface ClassificationModel

Parameters:

testData - array representing a single data point

Returns:

predicted category from the trained model