¿Cómo se puede visualizar el modelo de árbol de decisión en Spark (pyspark)?

Estoy tratando de visualizar la estructura del árbol de decisión en pyspark. Pero todas las herramientas son para datos. No pude encontrar ninguna para visualizar la estructura de un árbol. ¿O hay una manera en que puedo visualizar usando las reglas de toDebugString ?

He intentado hacer lo siguiente para crear una visualización:

  1. Parse Spark Decision Tree se envía a un formato JSON .
  2. Utilice el archivo JSON como entrada para una visualización D3.js

Para obtener más código, puede consultar mi prototipo en GitHub aquí .