public class NaiveBayesClassifier extends Object implements Serializable
Constructor and Description |
---|
NaiveBayesClassifier() |
Modifier and Type | Method and Description |
---|---|
org.apache.spark.api.java.JavaPairRDD<Double,Double> |
test(org.apache.spark.mllib.classification.NaiveBayesModel naiveBayesModel,
org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> test)
This method applies a naive bayes model to a given dataset
|
org.apache.spark.mllib.classification.NaiveBayesModel |
train(org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> trainingData,
double lambda)
This method trains a naive bayes model
|
public org.apache.spark.mllib.classification.NaiveBayesModel train(org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> trainingData, double lambda)
trainingData
- Training dataset as a JavaRDD of labeled pointslambda
- Lambda parameterpublic org.apache.spark.api.java.JavaPairRDD<Double,Double> test(org.apache.spark.mllib.classification.NaiveBayesModel naiveBayesModel, org.apache.spark.api.java.JavaRDD<org.apache.spark.mllib.regression.LabeledPoint> test)
naiveBayesModel
- Naive bayes modeltest
- Testing dataset as a JavaRDD of labeled pointsCopyright © 2015 WSO2, Inc.. All Rights Reserved.