ml.combust.mleap.core.classification

NaiveBayesModel

case class NaiveBayesModel(numFeatures: Int, numClasses: Int, pi: Vector, theta: Matrix, modelType: ModelType) extends ProbabilisticClassificationModel with Model with Product with Serializable

numFeatures

number of features in feature vector

numClasses

number of labels or labels to classify predictions into

pi

log of class priors

theta

log of class conditional probabilities

Annotations
@SparkCode()
Linear Supertypes
Serializable, Serializable, Product, Equals, ProbabilisticClassificationModel, ClassificationModel, Model, AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. NaiveBayesModel
  2. Serializable
  3. Serializable
  4. Product
  5. Equals
  6. ProbabilisticClassificationModel
  7. ClassificationModel
  8. Model
  9. AnyRef
  10. Any
  1. Hide All
  2. Show all
Learn more about member selection
Visibility
  1. Public
  2. All

Instance Constructors

  1. new NaiveBayesModel(numFeatures: Int, numClasses: Int, pi: Vector, theta: Matrix, modelType: ModelType)

    numFeatures

    number of features in feature vector

    numClasses

    number of labels or labels to classify predictions into

    pi

    log of class priors

    theta

    log of class conditional probabilities

Value Members

  1. final def !=(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  2. final def !=(arg0: Any): Boolean

    Definition Classes
    Any
  3. final def ##(): Int

    Definition Classes
    AnyRef → Any
  4. final def ==(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  5. final def ==(arg0: Any): Boolean

    Definition Classes
    Any
  6. def apply(features: Vector): Double

    Alias for ml.combust.mleap.core.classification.ClassificationModel#predict.

    features

    feature vector

    returns

    prediction

    Definition Classes
    ClassificationModel
  7. final def asInstanceOf[T0]: T0

    Definition Classes
    Any
  8. def clone(): AnyRef

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  9. final def eq(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  10. def finalize(): Unit

    Attributes
    protected[java.lang]
    Definition Classes
    AnyRef
    Annotations
    @throws( classOf[java.lang.Throwable] )
  11. final def getClass(): Class[_]

    Definition Classes
    AnyRef → Any
  12. def inputSchema: StructType

  13. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  14. val modelType: ModelType

  15. final def ne(arg0: AnyRef): Boolean

    Definition Classes
    AnyRef
  16. final def notify(): Unit

    Definition Classes
    AnyRef
  17. final def notifyAll(): Unit

    Definition Classes
    AnyRef
  18. val numClasses: Int

    number of labels or labels to classify predictions into

    number of labels or labels to classify predictions into

    Definition Classes
    NaiveBayesModelProbabilisticClassificationModel
  19. val numFeatures: Int

    number of features in feature vector

    number of features in feature vector

    Definition Classes
    NaiveBayesModelProbabilisticClassificationModel
  20. def outputSchema: StructType

  21. val pi: Vector

    log of class priors

  22. def predict(features: Vector): Double

    Predict class based on feature vector.

    Predict class based on feature vector.

    features

    feature vector

    returns

    predicted class or probability

    Definition Classes
    ProbabilisticClassificationModelClassificationModel
  23. def predictProbabilities(features: Vector): Vector

  24. def predictRaw(raw: Vector): Vector

  25. def predictWithProbability(features: Vector): (Double, Double)

  26. def probabilityToPrediction(probability: Vector): Double

  27. def probabilityToPredictionIndex(probability: Vector): Int

  28. def rawToPrediction(raw: Vector): Double

  29. def rawToProbability(raw: Vector): Vector

  30. def rawToProbabilityInPlace(raw: Vector): Vector

  31. final def synchronized[T0](arg0: ⇒ T0): T0

    Definition Classes
    AnyRef
  32. val theta: Matrix

    log of class conditional probabilities

  33. def thresholds: Option[Array[Double]]

  34. final def wait(): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  35. final def wait(arg0: Long, arg1: Int): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )
  36. final def wait(arg0: Long): Unit

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from Serializable

Inherited from Serializable

Inherited from Product

Inherited from Equals

Inherited from ClassificationModel

Inherited from Model

Inherited from AnyRef

Inherited from Any

Ungrouped