python - Scikit Learn - 决策树 - 每条记录结果的视觉表示

Question

我有一个工作决策树，还有一个来 self 的训练数据的工作 .Dot 输出文件。我想做的是输出每条记录的决策树过程。我的意思是，我想向用户显示每条记录达到决定结果所遵循的路径，而不是作为具有单个分支的树甚至作为文本。这可能吗？

Answer 1

您可以使用训练树的 decision_path 方法 ( here ) 来查看路径。但是，您可能只需要 apply 方法 ( here )。 apply 方法显示叶节点。由于到达每片叶子的方法只有一种，即了解叶子并查看树，即可推断出路径。

您想要如何显示它取决于您和用户。表格、可视化等。