Re: [scikit-learn] Issue with DecisionTreeClassifier

2016-08-30 Thread Andy
On 08/29/2016 11:22 PM, Nelson Liu wrote: 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. well it's also the weighted number of samples of each class, right? __

Re: [scikit-learn] Issue with DecisionTreeClassifier

2016-08-29 Thread Joel Nothman
Or just running estimator.tree_.apply(X_train) and inferring from there. On 30 August 2016 at 13:22, Nelson Liu wrote: > 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

Re: [scikit-learn] Issue with DecisionTreeClassifier

2016-08-29 Thread Nelson Liu
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 acc

Re: [scikit-learn] Issue with DecisionTreeClassifier

2016-08-29 Thread Ibrahim Dalal via scikit-learn
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 wrote: > > > On 08/28/2016 03:23 PM

Re: [scikit-learn] Issue with DecisionTreeClassifier

2016-08-29 Thread Andreas Mueller
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. _

Re: [scikit-learn] Issue with DecisionTreeClassifier

2016-08-29 Thread Ibrahim Dalal via scikit-learn
Thanks Nelson. Is there any way to access number of training samples in a node? Thanks On Mon, Aug 29, 2016 at 8:53 PM, Nelson Liu wrote: > Hi, > Yes, it's estimator.tree_.impurity > > Nelson > > On Mon, Aug 29, 2016, 09:18 Ibrahim Dalal via scikit-learn < > scikit-learn@python.org> wrote: > >>

Re: [scikit-learn] Issue with DecisionTreeClassifier

2016-08-29 Thread Nelson Liu
Hi, Yes, it's estimator.tree_.impurity Nelson On Mon, Aug 29, 2016, 09:18 Ibrahim Dalal via scikit-learn < scikit-learn@python.org> wrote: > Hi, > > Is there a way to extract impurity value of a node in > DecisionTreeClassifier? I am able to get this value in graph (using > export_grapgviz), but

Re: [scikit-learn] Issue with DecisionTreeClassifier

2016-08-29 Thread Ibrahim Dalal via scikit-learn
Hi, Is there a way to extract impurity value of a node in DecisionTreeClassifier? I am able to get this value in graph (using export_grapgviz), but can't figure out how to get this value in my code. Is there any attribute similar to estimator.tree_.children_left? Thanks On Mon, Aug 29, 2016 at 1

Re: [scikit-learn] Issue with DecisionTreeClassifier

2016-08-28 Thread Nelson Liu
Oops, phone removed the underscore between the two words of the variable name but I think you get the point. Nelson On Sun, Aug 28, 2016, 13:12 Ibrahim Dalal via scikit-learn < scikit-learn@python.org> wrote: > Dear Developers, > > DecisionTreeClassifier.decision_path() as used here > http://sci

Re: [scikit-learn] Issue with DecisionTreeClassifier

2016-08-28 Thread Nelson Liu
That should be: node indicator = estimator.tree_.decision_path(X_test) PR welcome :) On Sun, Aug 28, 2016, 13:12 Ibrahim Dalal via scikit-learn < scikit-learn@python.org> wrote: > Dear Developers, > > DecisionTreeClassifier.decision_path() as used here > http://scikit-learn.org/dev/auto_examples