estimator.tree_.value gives the constant prediction of the tree at each node. Think of it as what the tree would output if that node was a leaf.

I don't think we have a readily available way of checking the number of training samples of each class in a given tree node. The closest thing easily accessible is estimator.tree_.n_node_samples. Getting finer-grained counts of the number of samples in each class would require modifying the source code, I think.

On Mon, Aug 29, 2016 at 8:06 PM Ibrahim Dalal via scikit-learn <scikit-learn@python.org> wrote:
Hi,

What does the estimator.tree_.value array represent? I looked up the source code but not able to get what it is. I am interested in the number of training samples of each class in a given tree node.

Thanks

On Mon, Aug 29, 2016 at 9:22 PM, Andreas Mueller <t3kcit@gmail.com> wrote:


On 08/28/2016 03:23 PM, Nelson Liu wrote:
That should be:
node indicator = estimator.tree_.decision_path(X_test)

PR welcome :)
Was there a reason not to make this a "plot" example?
Would it take too long? Not having run examples by CI is a pretty big maintenance burden.

_______________________________________________
scikit-learn mailing list
scikit-learn@python.org
https://mail.python.org/mailman/listinfo/scikit-learn


_______________________________________________
scikit-learn mailing list
scikit-learn@python.org
https://mail.python.org/mailman/listinfo/scikit-learn