AttributeError: 'DecisionTreeClassifier' object has no attribute 'plot_tree'
时间: 2023-06-22 12:19:13 浏览: 1295
这个错误通常是因为您正在使用的是较早版本的Scikit-learn库,该库不支持plot_tree方法。您可以尝试更新Scikit-learn库以解决此问题。
如果更新库不可行,您也可以尝试使用export_graphviz方法来可视化决策树。您可以使用export_graphviz方法导出决策树的Graphviz表示,然后使用Graphviz软件包将其转换为可视化的图形。以下是一个示例代码片段:
```python
from sklearn.tree import export_graphviz
import graphviz
# 创建决策树模型
tree_clf = DecisionTreeClassifier()
# 训练模型
# 导出Graphviz表示
dot_data = export_graphviz(tree_clf, out_file=None,
feature_names=feature_names,
class_names=target_names,
filled=True, rounded=True,
special_characters=True)
# 将Graphviz表示转换为图形
graph = graphviz.Source(dot_data)
graph.render("decision_tree") # 将图形保存到文件中
graph.view() # 在窗口中显示图形
```
请注意,您需要安装Graphviz软件包才能使用此方法。您可以从Graphviz官方网站下载并安装该软件包。
阅读全文