org.apache.spark.mllib.classification

Class NaiveBayes

java.lang.Object

org.apache.spark.mllib.classification.NaiveBayes

All Implemented Interfaces:

java.io.Serializable, Logging

public class NaiveBayes
extends java.lang.Object
implements scala.Serializable, Logging

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
NaiveBayes()
NaiveBayes(double lambda)

Method Summary

Methods

Modifier and Type	Method and Description
static java.lang.String	Bernoulli()
double	getLambda()
java.lang.String	getModelType()
static java.lang.String	Multinomial()
NaiveBayesModel	run(RDD<LabeledPoint> data)
NaiveBayes	setLambda(double lambda)
NaiveBayes	setModelType(java.lang.String modelType)
static scala.collection.immutable.Set<java.lang.String>	supportedModelTypes()
static NaiveBayesModel	train(RDD<LabeledPoint> input)
static NaiveBayesModel	train(RDD<LabeledPoint> input, double lambda)
static NaiveBayesModel	train(RDD<LabeledPoint> input, double lambda, java.lang.String modelType)

Methods inherited from class java.lang.Object

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

Methods inherited from interface org.apache.spark.Logging

initializeIfNecessary, initializeLogging, isTraceEnabled, log_, log, logDebug, logDebug, logError, logError, logInfo, logInfo, logName, logTrace, logTrace, logWarning, logWarning

Constructor Detail

NaiveBayes

public NaiveBayes(double lambda)

NaiveBayes

public NaiveBayes()

Method Detail

Multinomial

public static java.lang.String Multinomial()

Bernoulli

public static java.lang.String Bernoulli()

supportedModelTypes

public static scala.collection.immutable.Set<java.lang.String> supportedModelTypes()

train

public static NaiveBayesModel train(RDD<LabeledPoint> input)

train

public static NaiveBayesModel train(RDD<LabeledPoint> input,
                    double lambda)

train

public static NaiveBayesModel train(RDD<LabeledPoint> input,
                    double lambda,
                    java.lang.String modelType)

setLambda

public NaiveBayes setLambda(double lambda)

getLambda

public double getLambda()

setModelType

public NaiveBayes setModelType(java.lang.String modelType)

getModelType

public java.lang.String getModelType()

run

public NaiveBayesModel run(RDD<LabeledPoint> data)