Mac安装画图工具pydotplus

原创于 2018-12-19 15:26:42 发布 · 946 阅读

0 ·

CC 4.0 BY-SA版权

机器学习专栏收录该内容

26 篇文章

订阅专栏

本文详细介绍了如何在Mac环境下安装Graphviz和Pydotplus，以便于在Python中使用Scikit-learn库进行决策树模型的可视化。通过具体的代码示例，展示了从数据准备、模型训练到最终生成决策树图像的全过程。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

Mac下安装graphviz

终端执行

brew install graphviz

然后安装

pip install pydotplus

到 notebook 里测试下

mport numpy as np
import pandas as pd
from sklearn.tree import DecisionTreeClassifier
# 仍然使用自带的iris数据
from sklearn.datasets import load_iris
iris = load_iris()
X = pd.DataFrame(iris.data[:, :-1])
Y = pd.DataFrame(iris.target)
model = DecisionTreeClassifier(criterion="entropy").fit(X,Y)
model.predict(X)
model
from IPython.display import Image  
from sklearn import tree
import pydotplus 
dot_data = tree.export_graphviz(model, out_file=None, 
                         filled=True, rounded=True,  
                         special_characters=True)  
graph = pydotplus.graph_from_dot_data(dot_data)  
Image(graph.create_png())