export_graphviz决策树可视化报错Tree not estimator instance

2019独角兽企业重金招聘Python工程师标准>>> hot3.png

报错信息

C:\Python27\python.exe D:/code/ML5/part08/xin/BugTest.py
Traceback (most recent call last):
  File "D:/code/ML5/part08/xin/BugTest.py", line 28, in 
    class_names=["is_date:no","is_date:yes"],filled=True,rounded=True)
  File "C:\Python27\lib\site-packages\sklearn\tree\export.py", line 392, in export_graphviz
    check_is_fitted(decision_tree, 'tree_')
  File "C:\Python27\lib\site-packages\sklearn\utils\validation.py", line 762, in check_is_fitted
    raise TypeError("%s is not an estimator instance." % (estimator))
TypeError:  is not an estimator instance.

出问题的代码

dtc.fit(X_train,y_train)

#6.可视化
from sklearn.tree import export_graphviz
export_graphviz(dtc.tree_,out_file="tan1.dot",)

这段代码来自网络,然后就看了下export_graphviz源码中的示例,发现了不同之处。

export_graphviz决策树可视化报错Tree not estimator instance_第1张图片

上文中的export_graphviz方法,参数是 dtc.tree_ ,这里取不到这个值。 需要给dtc.fit(X_train,y_train)赋一个返回值。

改成这样

dtc_tree=dtc.fit(X_train,y_train)

#6.可视化
from sklearn.tree import export_graphviz
export_graphviz(dtc_tree,out_file="tan1.dot",)

graphviz的使用

##1.下载安装graphviz 地址:https://graphviz.gitlab.io/download/

2.配置环境变量

在path中添加:C:\Program Files (x86)\Graphviz2.38\bin

3.将dot文件转成图片

在dot文件目录打开命令行,输入命令:

PS D:\code\jupyter\item-ratings> dot -Tpng digits.dot -o item-ratings.png
PS D:\code\jupyter\item-ratings>

转载于:https://my.oschina.net/datadev/blog/1825811

你可能感兴趣的:(export_graphviz决策树可视化报错Tree not estimator instance)