diff options
author | Sean Owen <sowen@cloudera.com> | 2015-07-14 22:44:54 +0100 |
---|---|---|
committer | Sean Owen <sowen@cloudera.com> | 2015-07-14 22:44:54 +0100 |
commit | 740b034f1ca885a386f5a9ef7e0c81c714b047ff (patch) | |
tree | 144d62439bad5c1cdc406ca843b41472941c7cef /repl/src/main/scala | |
parent | 4b5cfc988f23988c2334882a255d494fc93d252e (diff) | |
download | spark-740b034f1ca885a386f5a9ef7e0c81c714b047ff.tar.gz spark-740b034f1ca885a386f5a9ef7e0c81c714b047ff.tar.bz2 spark-740b034f1ca885a386f5a9ef7e0c81c714b047ff.zip |
[SPARK-4362] [MLLIB] Make prediction probability available in NaiveBayesModel
Add predictProbabilities to Naive Bayes, return class probabilities.
Continues https://github.com/apache/spark/pull/6761
Author: Sean Owen <sowen@cloudera.com>
Closes #7376 from srowen/SPARK-4362 and squashes the following commits:
23d5a76 [Sean Owen] Fix model.labels -> model.theta
95d91fb [Sean Owen] Check that predicted probabilities sum to 1
b32d1c8 [Sean Owen] Add predictProbabilities to Naive Bayes, return class probabilities
Diffstat (limited to 'repl/src/main/scala')
0 files changed, 0 insertions, 0 deletions