如何显示测试样本的决策树路径？

提问者：小点点

如何显示测试样本的决策树路径？

我正在使用从Scikit学习决策树分类器来分类一些多类数据。我发现了许多描述如何显示决策树路径的帖子，比如这里、这里和这里。然而，它们都描述了如何为训练的数据显示树。这是有道理的，因为export_graphviz只需要一个拟合的模型。

我的问题是如何可视化测试样本上的树（最好是通过export\u graphviz）。即，使用clf拟合模型后。拟合（X[train]，y[train]），然后通过clf预测测试数据的结果。predict（X[test]），我想可视化用于预测样本的决策路径X[test]。有办法吗？

编辑：

我看到可以使用decision_path打印路径。如果有一种方法可以得到一个DOT输出作为export_graphviz来显示它，那就太好了。

共1个答案

匿名用户

为了获得决策树中特定样本的路径，您可以使用decision_path。它返回一个稀疏矩阵，其中包含所提供样本的决策路径。

然后，可以使用这些决策路径对通过pydot生成的树进行着色/标记。这需要覆盖颜色和标签（这会导致一些难看的代码）。

笔记

decision\u path可以从训练集或新值中提取样本
您可以随意选择颜色，并根据样本数量或可能需要的任何其他可视化更改颜色

实例

在下面的示例中，访问的节点为绿色，所有其他节点为白色。

import pydotplus
from sklearn.datasets import load_iris
from sklearn import tree

clf = tree.DecisionTreeClassifier(random_state=42)
iris = load_iris()

clf = clf.fit(iris.data, iris.target)

dot_data = tree.export_graphviz(clf, out_file=None,
                                feature_names=iris.feature_names,
                                class_names=iris.target_names,
                                filled=True, rounded=True,
                                special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data)

# empty all nodes, i.e.set color to white and number of samples to zero
for node in graph.get_node_list():
    if node.get_attributes().get('label') is None:
        continue
    if 'samples = ' in node.get_attributes()['label']:
        labels = node.get_attributes()['label'].split('<br/>')
        for i, label in enumerate(labels):
            if label.startswith('samples = '):
                labels[i] = 'samples = 0'
        node.set('label', '<br/>'.join(labels))
        node.set_fillcolor('white')

samples = iris.data[129:130]
decision_paths = clf.decision_path(samples)

for decision_path in decision_paths:
    for n, node_value in enumerate(decision_path.toarray()[0]):
        if node_value == 0:
            continue
        node = graph.get_node(str(n))[0]            
        node.set_fillcolor('green')
        labels = node.get_attributes()['label'].split('<br/>')
        for i, label in enumerate(labels):
            if label.startswith('samples = '):
                labels[i] = 'samples = {}'.format(int(label.split('=')[1]) + 1)

        node.set('label', '<br/>'.join(labels))

filename = 'tree.png'
graph.write_png(filename)