python - 如何从决策树的 x_train 预测位置获取叶子的节点号？

Question

我有一个训练有素的决策树。当我输入一个特征向量来预测时，我想知道它是从哪个决策路径预测的，或者是在新特征属于树的哪一片叶子下预测的。

我正在使用Python的Sklearn的决策树实现。

Answer 1

有一种方法可以使用类的 decision_path 方法来访问树中的决策路径。

示例:

from sklearn.ensemble import RandomForestClassifier
from sklearn.datasets import load_iris 
import numpy as np

data = load_iris()

x = data.data
y = data.target

clf = RandomForestClassifier()

clf.fit(x,y)

clf.decision_path(x)

结果:

(<150x140 sparse matrix of type '<type 'numpy.int64'>'
with 5406 stored elements in Compressed Sparse Row format>, array([  0,  13,  
26,  41,  54,  71,  86,  97, 106, 119, 140]))