GitHub user leahmcguire opened a pull request:
https://github.com/apache/spark/pull/4087
[SPARK-4894][mllib] Added Bernoulli option to NaiveBayes model in mllib
Added optional model type parameter for NaiveBayes training. Can be either
Multinomial or Bernoulli.
When Bernoulli is given the Bernoulli smoothing is used for fitting and for
prediction as per:
http://nlp.stanford.edu/IR-book/html/htmledition/the-bernoulli-model-1.html.
Default for model is original Multinomial fit and predict.
Added additional testing for Bernoulli and Multinomial models.
You can merge this pull request into a Git repository by running:
$ git pull https://github.com/leahmcguire/spark master
Alternatively you can review and apply these changes as the patch at:
https://github.com/apache/spark/pull/4087.patch
To close this pull request, make a commit to your master/trunk branch
with (at least) the following in the commit message:
This closes #4087
----
commit ce73c63e8bac40b02ae0a8147c3b424783f6094a
Author: leahmcguire <[email protected]>
Date: 2015-01-16T16:06:06Z
added Bernoulli option to niave bayes model in mllib, added optional model
type parameter for training. When Bernoulli is given the Bernoulli smoothing is
used for fitting and for prediction
http://nlp.stanford.edu/IR-book/html/htmledition/the-bernoulli-model-1.html
----
---
If your project is set up for it, you can reply to this email and have your
reply appear on GitHub as well. If your project does not have this feature
enabled and wishes so, or if the feature is enabled but not working, please
contact infrastructure at [email protected] or file a JIRA ticket
with INFRA.
---
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
