Github user jkbradley commented on the pull request:
https://github.com/apache/spark/pull/3583#issuecomment-66348700
@dikejiang Thanks for the PR! I'm wondering if you'd be interested in a
more general API. In the new experimental ML package, I have a PR
[https://www.github.com/apache/spark/pull/3637] which introduces a few
prediction methods, one of which is:
```
def predictRaw(features: Vector): Vector // for each label, predict a
confidence
```
What do you think of using this instead of only predicting the top label's
weight? Eventually, confidence predictions could be improved by incorporating
each tree's confidence in its prediction (rather than having each tree simply
vote for a single label, as is done now).
---
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]
