如何将Spark(pyspark)中的决策树模型可视化？

Question

我想在pyspark中实现决策树结构的可视化。但是所有的工具都是针对数据的。我找不到任何可视化树结构的工具。或者有什么方法可以让我使用来自于 toDebugString?

Answer 1

我已经尝试做以下工作，以创建一个可视化的。

1. 将Spark决策树的输出解析为 JSON 格式。
2. 使用 JSON 文件作为输入到 D3.js 的可视化。

更多代码可以参考我在GitHub的原型。此处.