win64 python pydot安装配置

本文主要是win10，下面anaconda安装pydot以及 .dot文件转化为png。

---

1. 安装pydot

pip install pydot

---

2. 安装GraphViz

链接如下：
http://www.graphviz.org/Download_windows.php

下载完成点击安装，选择好安装目录之后即可。最终将安装路径添加到，环境变量中。

3. 重启pycharm

from sklearn.datasets import load_irisfrom sklearn import treefrom sklearn.tree import DecisionTreeClassifier, export_graphvizimport subprocessimport pydotclf = tree.DecisionTreeClassifier()iris = load_iris()clf = clf.fit(iris.data, iris.target)tree.export_graphviz(clf, out_file='tree.dot')(graph,) = pydot.graph_from_dot_file('tree.dot')graph.write_png('somefile.png')

生成下面的tree

---

from sklearn.datasets import load_irisfrom sklearn import treefrom sklearn.tree import DecisionTreeClassifier, export_graphvizimport subprocessimport pydotdef visualize_tree(tree):    """Create tree png using graphviz.    Args    ----    tree -- scikit-learn DecsisionTree.    feature_names -- list of feature names.    """    with open("tree.dot", 'w') as f:        export_graphviz(tree, out_file=f,                        feature_names=iris.feature_names)    command = ["dot", "-Tpng", "tree.dot", "-o", "dt.png"]    try:        subprocess.check_call(command)    except:        exit("Could not run dot, ie graphviz, to "             "produce visualization")clf = tree.DecisionTreeClassifier()iris = load_iris()clf = clf.fit(iris.data, iris.target)tree.export_graphviz(clf, out_file='tree.dot')visualize_tree(clf)

这个可视化，添加了属性，看起来更直观

转载注明出处，并在下面留言！！！