Packages

  • package root
    Definition Classes
    root
  • package org
    Definition Classes
    root
  • package apache
    Definition Classes
    org
  • package spark

    Core Spark functionality.

    Core Spark functionality. org.apache.spark.SparkContext serves as the main entry point to Spark, while org.apache.spark.rdd.RDD is the data type representing a distributed collection, and provides most parallel operations.

    In addition, org.apache.spark.rdd.PairRDDFunctions contains operations available only on RDDs of key-value pairs, such as groupByKey and join; org.apache.spark.rdd.DoubleRDDFunctions contains operations available only on RDDs of Doubles; and org.apache.spark.rdd.SequenceFileRDDFunctions contains operations available on RDDs that can be saved as SequenceFiles. These operations are automatically available on any RDD of the right type (e.g. RDD[(Int, Int)] through implicit conversions.

    Java programmers should reference the org.apache.spark.api.java package for Spark programming APIs in Java.

    Classes and methods marked with Experimental are user-facing features which have not been officially adopted by the Spark project. These are subject to change or removal in minor releases.

    Classes and methods marked with Developer API are intended for advanced users want to extend Spark through lower level interfaces. These are subject to changes or removal in minor releases.

    Definition Classes
    apache
  • package mllib

    RDD-based machine learning APIs (in maintenance mode).

    RDD-based machine learning APIs (in maintenance mode).

    The spark.mllib package is in maintenance mode as of the Spark 2.0.0 release to encourage migration to the DataFrame-based APIs under the org.apache.spark.ml package. While in maintenance mode,

    • no new features in the RDD-based spark.mllib package will be accepted, unless they block implementing new features in the DataFrame-based spark.ml package;
    • bug fixes in the RDD-based APIs will still be accepted.

    The developers will continue adding more features to the DataFrame-based APIs in the 2.x series to reach feature parity with the RDD-based APIs. And once we reach feature parity, this package will be deprecated.

    Definition Classes
    spark
    See also

    SPARK-4591 to track the progress of feature parity

  • package classification
    Definition Classes
    mllib
  • ClassificationModel
  • LogisticRegressionModel
  • LogisticRegressionWithLBFGS
  • LogisticRegressionWithSGD
  • NaiveBayes
  • NaiveBayesModel
  • SVMModel
  • SVMWithSGD
  • StreamingLogisticRegressionWithSGD

class NaiveBayesModel extends ClassificationModel with Serializable with Saveable

Model for Naive Bayes Classifiers.

Annotations
@Since("0.9.0")
Source
NaiveBayes.scala
Linear Supertypes
Saveable, ClassificationModel, Serializable, AnyRef, Any
Ordering
  1. Alphabetic
  2. By Inheritance
Inherited
  1. NaiveBayesModel
  2. Saveable
  3. ClassificationModel
  4. Serializable
  5. AnyRef
  6. Any
  1. Hide All
  2. Show All
Visibility
  1. Public
  2. Protected

Value Members

  1. val labels: Array[Double]
    Annotations
    @Since("1.0.0")
  2. val modelType: String
    Annotations
    @Since("1.4.0")
  3. val pi: Array[Double]
    Annotations
    @Since("0.9.0")
  4. def predict(testData: Vector): Double

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

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

    testData

    array representing a single data point

    returns

    predicted category from the trained model

    Definition Classes
    NaiveBayesModel → ClassificationModel
    Annotations
    @Since("1.0.0")
  5. def predict(testData: RDD[Vector]): RDD[Double]

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

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

    testData

    RDD representing data points to be predicted

    returns

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

    Definition Classes
    NaiveBayesModel → ClassificationModel
    Annotations
    @Since("1.0.0")
  6. def predict(testData: JavaRDD[Vector]): JavaRDD[Double]

    Predict values for examples stored in a JavaRDD.

    Predict values for examples stored in a JavaRDD.

    testData

    JavaRDD representing data points to be predicted

    returns

    a JavaRDD[java.lang.Double] where each entry contains the corresponding prediction

    Definition Classes
    ClassificationModel
    Annotations
    @Since("1.0.0")
  7. def predictProbabilities(testData: Vector): Vector

    Predict posterior class probabilities for a single data point using the model trained.

    Predict posterior class probabilities for a single data point using the model trained.

    testData

    array representing a single data point

    returns

    predicted posterior class probabilities from the trained model, in the same order as class labels

    Annotations
    @Since("1.5.0")
  8. def predictProbabilities(testData: RDD[Vector]): RDD[Vector]

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

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

    testData

    RDD representing data points to be predicted

    returns

    an RDD[Vector] where each entry contains the predicted posterior class probabilities, in the same order as class labels

    Annotations
    @Since("1.5.0")
  9. def save(sc: SparkContext, path: String): Unit

    Save this model to the given path.

    Save this model to the given path.

    This saves:

    • human-readable (JSON) model metadata to path/metadata/
    • Parquet formatted data to path/data/

    The model may be loaded using Loader.load.

    sc

    Spark context used to save model data.

    path

    Path specifying the directory in which to save this model. If the directory already exists, this method throws an exception.

    Definition Classes
    NaiveBayesModel → Saveable
    Annotations
    @Since("1.3.0")
  10. val theta: Array[Array[Double]]
    Annotations
    @Since("0.9.0")