[SPARK-6884] Random forest: predict class probabilities

• Type: Sub-task
• Status: Closed
• Priority: Major
• Resolution: Duplicate
• Component/s: ML
• Labels: prediction probability randomforest tree
• Target Version/s: 1.5.0
• Flags: Important
• Assignee: Unassigned
• Reporter: Max Kaznady
• Created: 13/Apr/15 19:00
• Updated: 12/Aug/15 21:28
• Resolved: 28/Jul/15 19:41

Description

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.

Issue Links

• is superceded by: SPARK-9016 Make random forest classifier extend Classifier abstraction
• relates to: SPARK-9904 User guide for ML tree algorithms