ml.combust.mleap.core.classification

NaiveBayesModel

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

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
Product, Equals, Serializable, Serializable, ProbabilisticClassificationModel, ClassificationModel, AnyRef, Any
Ordering
  1. Alphabetic
  2. By inheritance
Inherited
  1. NaiveBayesModel
  2. Product
  3. Equals
  4. Serializable
  5. Serializable
  6. ProbabilisticClassificationModel
  7. ClassificationModel
  8. AnyRef
  9. 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. final def isInstanceOf[T0]: Boolean

    Definition Classes
    Any
  13. val modelType: ModelType

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

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

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

    Definition Classes
    AnyRef
  17. val numClasses: Int

    number of labels or labels to classify predictions into

    number of labels or labels to classify predictions into

    Definition Classes
    NaiveBayesModelProbabilisticClassificationModel
  18. val numFeatures: Int

    number of features in feature vector

  19. val pi: Vector

    log of class priors

  20. 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
  21. def predictProbabilities(features: Vector): Vector

  22. def predictRaw(raw: Vector): Vector

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

  24. def probabilityToPrediction(probability: Vector): Double

  25. def rawToPrediction(raw: Vector): Double

  26. def rawToProbability(raw: Vector): Vector

  27. def rawToProbabilityInPlace(raw: Vector): Vector

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

    Definition Classes
    AnyRef
  29. val theta: Matrix

    log of class conditional probabilities

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

  31. final def wait(): Unit

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

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

    Definition Classes
    AnyRef
    Annotations
    @throws( ... )

Inherited from Product

Inherited from Equals

Inherited from Serializable

Inherited from Serializable

Inherited from ClassificationModel

Inherited from AnyRef

Inherited from Any

Ungrouped