python 遍历决策树的节点_为决策树中的每个数据点查找对应的叶节点(scikitlearn)...

我使用的是python3.4中scikit learn包中的决策树分类器,我希望获得每个输入数据点对应的叶节点id。在

例如,我的输入可能如下所示:array([[ 5.1, 3.5, 1.4, 0.2],

[ 4.9, 3. , 1.4, 0.2],

[ 4.7, 3.2, 1.3, 0.2]])

假设对应的叶节点分别为16、5和45。我希望我的输出是:

^{pr2}$

我已经读完了scikit学习邮件列表和关于SF的相关问题,但是我还是不能让它发挥作用。这是我在邮件列表上找到的一些提示,但仍然不起作用。在

最后,我只想有一个GetLeafNode(clf,X_valida)函数,这样它的输出就是相应叶节点的列表。下面是重现我收到的错误的代码。所以,任何建议都将不胜感激。在from sklearn.datasets import load_iris

from sklearn import tree

# load data and divide it to train and validation

iris = load_iris()

num_train = 100

X_train = iris.data[:num_train,:]

X_valida = iris.data[num_train:,:]

y_train = iris.target[:num_train]

y_valida = iris.target[num_train:]

# fit the decision tree using the train data set

clf = tree.DecisionTreeClassifier()<

  • 0
    点赞
  • 0
    收藏
    觉得还不错? 一键收藏
  • 0
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值