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

报错信息

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 <module>
    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: <sklearn.tree._tree.Tree object at 0x0000000018479D98> 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方法,参数是 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",)

猜你喜欢

转载自my.oschina.net/datadev/blog/1825811