如何将export_graphviz结果适合浏览器

时间:2018-07-03 02:00:05

标签: python tree scikit-learn graphviz

我正在使用export_graphviz绘制分类树:

from sklearn.datasets import load_iris
from sklearn import tree
iris = load_iris()
clf = tree.DecisionTreeClassifier()
clf = clf.fit(iris.data, iris.target)

import graphviz 
dot_data = tree.export_graphviz(clf, out_file=None) 
graph = graphviz.Source(dot_data) 
graph.render("iris") 

dot_data = tree.export_graphviz(clf, out_file=None, 
                         feature_names=iris.feature_names,  
                         class_names=iris.target_names,  
                         filled=True, rounded=True,  
                         special_characters=True  )  
graph = graphviz.Source(dot_data)  
graph

但是它正在像这样从浏览器窗口中消失: Does not fint into the window

如何在无需滚动的情况下将图形拟合到jupyter窗口中?

我没有找到任何解决方案。但是,我正在使用一个技巧来解决此问题。我将jupyter的width属性更改为100%,所以当我缩小时,宽度在高度变化时保持不变,这样我就可以看到整个树。

from IPython.core.display import display, HTML
display(HTML("<style>.container { width:100% !important; }</style>"))

如果有人能给我一个真正的解决方案,我将非常感谢。

0 个答案:

没有答案