2019独角兽企业重金招聘Python工程师标准>>>
报错信息
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方法,参数是 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>