Spark（pyspark）中的决策树模型如何可视化？答案

【问题标题】：How can decision tree model in Spark (pyspark) be visualized?Spark（pyspark）中的决策树模型如何可视化？
【发布时间】：2016-07-08 11:00:33
【问题描述】：

我正在尝试在 pyspark 中可视化决策树结构。但是所有的工具都是为了数据。我找不到任何可视化树结构的东西。或者有没有一种方法可以使用toDebugString 中的规则进行可视化？

【问题讨论】：

【解决方案1】：

我已尝试执行以下操作以创建可视化：

更多代码可以参考我在 GitHub 上的原型here。

【讨论】：

【解决方案2】：

我也是一名大数据/ml 工程师，我有很多时间需要可视化 Spark 的决策树。

我们刚刚发布了dtreeviz 1.1，其中包括对 Spark 的支持。您现在可以可视化树结构、叶子信息、预测路径等。只需查看此notebook 以获取更多示例。

【讨论】：