我正在使用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
如何在无需滚动的情况下将图形拟合到jupyter窗口中?
我没有找到任何解决方案。但是,我正在使用一个技巧来解决此问题。我将jupyter的width属性更改为100%,所以当我缩小时,宽度在高度变化时保持不变,这样我就可以看到整个树。
from IPython.core.display import display, HTML
display(HTML("<style>.container { width:100% !important; }</style>"))
如果有人能给我一个真正的解决方案,我将非常感谢。