Max Kaznady created SPARK-6884:
----------------------------------
Summary: random forest predict probabilities functionality (like
in sklearn)
Key: SPARK-6884
URL: https://issues.apache.org/jira/browse/SPARK-6884
Project: Spark
Issue Type: New Feature
Components: MLlib
Affects Versions: 1.4.0
Environment: cross-platform
Reporter: Max Kaznady
Currently, there is no way to extract the class probabilities from the
RandomForest classifier. I implemented a probability predictor by counting
votes from individual trees and adding up their votes for "1" and then dividing
by the total number of votes.
I opened this ticked to keep track of changes. Will update once I push my code
to master.
--
This message was sent by Atlassian JIRA
(v6.3.4#6332)
---------------------------------------------------------------------
To unsubscribe, e-mail: [email protected]
For additional commands, e-mail: [email protected]
