Package pyspark :: Package mllib :: Module classification :: Class NaiveBayesModel

Class NaiveBayesModel

object --+
         |
        NaiveBayesModel


Model for Naive Bayes classifiers.

Contains two parameters:
- pi: vector of logs of class priors (dimension C)
- theta: matrix of logs of class conditional probabilities (CxD)

>>> data = array([0.0, 0.0, 1.0, 1.0, 1.0, 0.0, 2.0, 1.0, 1.0]).reshape(3,3)
>>> model = NaiveBayes.train(sc.parallelize(data))
>>> model.predict(array([0.0, 1.0]))
0
>>> model.predict(array([1.0, 0.0]))
1

Instance Methods

__init__(self, pi, theta)
x.__init__(...) initializes x; see help(type(x)) for signature

source code

predict(self, x)
Return the most likely class for a data vector x

source code

Inherited from object: __delattr__, __format__, __getattribute__, __hash__, __new__, __reduce__, __reduce_ex__, __repr__, __setattr__, __sizeof__, __str__, __subclasshook__

Properties
Inherited from `object`: `__class__`

Method Details

init(self, pi, theta)
(Constructor)

source code

x.__init__(...) initializes x; see help(type(x)) for signature

Overrides: object.__init__: (inherited documentation)

Class NaiveBayesModel

__init__(self, pi, theta) (Constructor)

init(self, pi, theta)
(Constructor)