There is a JIRA...but not a PR yet. Here's the JIRA:
https://issues.apache.org/jira/browse/SPARK-3727
I'm not aware of current work on it, but I agree it would be nice to have!
Joseph
On Thu, Jan 22, 2015 at 2:50 AM, Sean Owen wrote:
> You are right that this isn't implemented. I presume you c
You are right that this isn't implemented. I presume you could propose
a PR for this. The impurity calculator implementations already receive
category counts. The only drawback I see is having to store N
probabilities at each leaf, not 1.
On Wed, Jan 21, 2015 at 3:36 PM, Zsolt Tóth wrote:
> Hi,
>
Hi,
I use DecisionTree for multi class classification.
I can get the probability of the predicted label for every node in the
decision tree from node.predict().prob(). Is it possible to retrieve or
count the probability of every possible label class in the node?
To be more clear:
Say in Node A the
