You are viewing a plain text version of this content. The canonical link for it is here.
Posted to issues@spark.apache.org by "Joseph K. Bradley (JIRA)" <ji...@apache.org> on 2015/07/20 19:20:05 UTC
[jira] [Issue Comment Deleted] (SPARK-6884) Random forest: predict
class probabilities
[ https://issues.apache.org/jira/browse/SPARK-6884?page=com.atlassian.jira.plugin.system.issuetabpanels:all-tabpanel ]
Joseph K. Bradley updated SPARK-6884:
-------------------------------------
Comment: was deleted
(was: Yep, it's been merged! I think this can initially aggregate votes from each tree, and then after [SPARK-6885] gets fixed, we can improve this using the soft (probability) predictions.)
> Random forest: predict class probabilities
> ------------------------------------------
>
> Key: SPARK-6884
> URL: https://issues.apache.org/jira/browse/SPARK-6884
> Project: Spark
> Issue Type: Sub-task
> Components: ML
> 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-unsubscribe@spark.apache.org
For additional commands, e-mail: issues-help@spark.apache.org