graphviz图片保存

dot_data = tree.export_graphviz(dt,out_file=None,feature_names=X.columns,filled=True,rounded=True,special_characters=True)
graph = graphviz.Source(dot_data,format='png')
graph.render('../trees/tree_%d' % i,cleanup=False)
graphviz.Source返回<class 'graphviz.files.Source'>结构,这个结构有个render方法,可以存储到本地,不加后缀直接存储的是png和dot文件,cleanup=False可以去掉dot文件,只保留图片。但是我遇到个小问题,图片中的中文无法正常显示,目前并没太多寻找解决方案,暂时需要不强烈

留言

熱門文章