tree.export_graphviz超list
时间: 2023-10-08 13:13:44 浏览: 87
解决使用export_graphviz可视化树报错的问题
我不太明白您的问题。不过,如果您是想将 `tree.export_graphviz` 输出的决策树图保存为图片而不是文本,可以使用 `pydotplus` 库将其转换为图片。下面是一个示例代码:
```python
from sklearn.datasets import load_iris
from sklearn.tree import DecisionTreeClassifier, export_graphviz
import pydotplus
from IPython.display import Image
iris = load_iris()
X = iris.data
y = iris.target
clf = DecisionTreeClassifier()
clf.fit(X, y)
dot_data = export_graphviz(clf, out_file=None, feature_names=iris.feature_names, class_names=iris.target_names, filled=True, rounded=True, special_characters=True)
graph = pydotplus.graph_from_dot_data(dot_data)
Image(graph.create_png())
```
这个代码会输出一个决策树图片。您可以通过调整 `Image()` 函数的参数来将其保存为文件。例如,将 `Image(graph.create_png())` 改为 `graph.write_png("tree.png")` 就会将图片保存为 "tree.png"。
阅读全文