[jira] [Updated] (SPARK-6884) random forest predict probabilities functionality (like in sklearn)
[ https://issues.apache.org/jira/browse/SPARK-6884?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel ] Joseph K. Bradley updated SPARK-6884: - Issue Type: Sub-task (was: New Feature) Parent: SPARK-3727 > random forest predict probabilities functionality (like in sklearn) > --- > > Key: SPARK-6884 > URL: https://issues.apache.org/jira/browse/SPARK-6884 > Project: Spark > Issue Type: Sub-task > Components: MLlib >Affects Versions: 1.3.0 > Environment: cross-platform >Reporter: Max Kaznady > Labels: prediction, probability, randomforest, tree > Original Estimate: 72h > Remaining Estimate: 72h > > 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: issues-unsubscr...@spark.apache.org For additional commands, e-mail: issues-h...@spark.apache.org
[jira] [Updated] (SPARK-6884) random forest predict probabilities functionality (like in sklearn)
[ https://issues.apache.org/jira/browse/SPARK-6884?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel ] Joseph K. Bradley updated SPARK-6884: - Affects Version/s: (was: 1.4.0) 1.3.0 > 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.3.0 > Environment: cross-platform >Reporter: Max Kaznady > Labels: prediction, probability, randomforest, tree > Original Estimate: 72h > Remaining Estimate: 72h > > 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: issues-unsubscr...@spark.apache.org For additional commands, e-mail: issues-h...@spark.apache.org
