رسم نمودار الگوریتم درخت تصمیم در پایتون

برای رسم نمودار درخت تصمیم ابتدا باید کتابخانه های زیر را فراخوانی کرد:

from sklearn.externals.six import StringIO  
from IPython.display import Image  
from sklearn.tree import export_graphviz
import pydotplus
from matplotlib.pyplot import plt 
plt.show()

نکته قابل توجه اینکه باید از اینجا فایل graphviz متناسب با سیستم عامل خود رو دریافت کنید. سپس روی سیستم خود نصب کنید. (مراحل نصب خیلی ساده هست و صرفا چند تا next پشت سر هم هست.) در ادامه باید تو مسیر نصب بروید و پوشه bin رو پیدا کنید و آدرسش رو کپی کنید. در ادامه مسیر زیر رو پیدا کنید:

Control Panel > System and Security > System > Advanced System Settings > Environment Variables > Path > Edit > New

و در نهایت آدرس پوشه bin رو در کادر زیر وارد کنید:

و حالا کد رو ادامه می دهیم:

import sklearn.datasets as datasets
import pandas as pd
iris=datasets.load_iris()
df=pd.DataFrame(iris.data, columns=iris.feature_names)
y=iris.target

در مورد مجموعه داده گل زنبق (iris) می تونید به اینجا مراجعه کنید.

from sklearn.tree import DecisionTreeClassifier
dtree=DecisionTreeClassifier()
dtree.fit(df,y)

و در ادامه:

dot_data = StringIO()

export_graphviz(dtree, out_file=dot_data,  
                filled=True, rounded=True,
                special_characters=True)

graph = pydotplus.graph_from_dot_data(dot_data.getvalue())  
Image(graph.create_png())

نمودار زیر رو می توانید مشاهده کنید:

منابع:

https://medium.com/@rnbrown/creating-and-visualizing-decision-trees-with-python-f8e8fa394176

https://stackoverflow.com/questions/28312534/graphvizs-executables-are-not-found-python-3-4

https://stackoverflow.com/questions/18438997/why-is-pydot-unable-to-find-graphvizs-executables-in-windows-8?noredirect=1&lq=1