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 /core/src/main/scala/org/apache/spark/executor/Executor.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 'core/src/main/scala/org/apache/spark/executor/Executor.scala')
0 files changed, 0 insertions, 0 deletions