决策树----iris案例（代码）

有个网站值得学习：http://cwiki.apachecn.org/pages/viewpage.action?pageId=10813980

from sklearn.datasets import load_iris
from sklearn import preprocessing
from sklearn.model_selection import train_test_split
from sklearn.tree import DecisionTreeClassifier
from sklearn.tree import export_graphviz
from sklearn.externals.six import StringIO
import pydotplus
from IPython.display import display,Image

iris=load_iris()
x=iris.data  #数据特征
y=iris.target#数据特征
x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=0.2,random_state=1)
scaler=preprocessing.StandardScaler().fit(x_train)
x1_train=scaler.transform(x_train)
x1_test=scaler.transform(x_test)

clf=DecisionTreeClassifier(criterion='entropy')
clf.fit(x_train,y_train)
y_pre=clf.predict(x1_test)
print(clf.score(x1_test,y_test))

dot_data=StringIO()
export_graphviz(clf,out_file=dot_data,
               feature_names=iris.feature_names,
               class_names=iris.target_names,
               filled=True,rounded=True,
               special_characters=True)
graph=pydotplus.graph_from_dot_data(dot_data.getvalue())
graph.write_png('D:\大二2\人工智能与金融\人工智能与金融应用_课内\iris.png')
display(Image(graph.create_png()))

决策树----iris案例（代码）

首先我们需要先安装Graphviz，安装后配置path环境（计算机—高级系统设置—环境变量----path）

注意：一定要有graphviz一定要有bin目录文件
安装好graphviz后，下载pydotplus
如果上面两步骤弄好后，运行还是会报GraphViz’s executables not found错误，可能是
（1）graphviz安装了但没有bin目录文件
（2）环境没配置好

决策树----iris案例（代码）

相关推荐