我正在做练习,在 jupyter 笔记本中使用 graphviz 创建决策树。然而,决策树过于宽泛。这是代码:
from sklearn.tree import export_graphviz
export_graphviz(tree, out_file="tree.dot", class_names=["malignant", "benign"],
feature_names=cancer.feature_names, impurity=False, filled=True)
with open("tree.dot") as f:
dot_graph = f.read()
display(graphviz.Source(dot_graph))
Run Code Online (Sandbox Code Playgroud)
我必须滚动才能看到决策树的左侧。我可以把宽度变小吗?如何?