如何在本地硬盘上保存"dtreeviz.trees“dtreeviz输出图

Question

以下是我的代码：

from sklearn import tree
from dtreeviz.trees import dtreeviz # will be used for tree visualization
from matplotlib import pyplot as plt
import graphviz

ax2 = dtreeviz(
 RFC1.estimators_[0],
 X_train1, y_train1, target_name='Mask',
 scale=1.1, class_names=['0','1'])
ax2

当我在做jupyter笔记本的时候，我一运行ax2就可以看到这个图表。我想把这个输出以PNG格式保存到我的本地系统上，但是我可以找到一种方法来做到这一点！

Answer 1

您可以将其保存为svg格式。

​

viz = dtreeviz(spark_dtree)

viz.save("/path/dtreeviz.svg")

​