使用export_graphviz生成决策树
from sklearn.tree import export_graphviz
export_graphviz(
tree_clf,
out_file=image_path("iris_tree.dot"),
impurity = True,
feature_names=iris.feature_names[2:],
class_names=iris.target_names,
rounded=True,
filled=True
)
报错
FileNotFoundError: [Errno 2] No such file or directory: '.\\images\\decision_trees\\iris_tree.dot'
暂时未找到如何解决方法,所以只能换一种方式
with open("iris_tree.dot", 'w') as f:
f = export_graphviz(tree_clf,
out_file=f,
feature_names=iris.feature_names[2:],
class_names=iris.target_names,
rounded = True,
filled= True )
OK,可以运行。