id3sklearn算法实现（西瓜数据集）

使用sklearn id3算法基于西瓜书上的西瓜集训练出模型&＃xff0c;效果&＃xff1a;

# 读取西瓜数据集
import numpy as np
import pandas as pd
df &＃61;pd.read_table(r&＃39;watermelon.txt&＃39;,encoding&＃61;&＃39;utf8&＃39;,delimiter&＃61;&＃39;,&＃39;,index_col&＃61;0)
df.head()
# 由于上面的数据中包含了中文汉字&＃xff0c;所以需要对数据进一步处理
&＃39;&＃39;&＃39;
属性&＃xff1a;
色泽 1-3代表 浅白 青绿 乌黑 根蒂 1-3代表 稍蜷 蜷缩 硬挺
敲声 1-3代表 清脆 浊响 沉闷 纹理 1-3代表 清晰 稍糊 模糊
脐部 1-3代表 平坦 稍凹 凹陷 触感 1-2代表 硬滑 软粘
标签&＃xff1a;
好瓜 1代表 是 0 代表 不是
&＃39;&＃39;&＃39;
df[&＃39;色泽&＃39;]&＃61;df[&＃39;色泽&＃39;].map({&＃39;浅白&＃39;:1,&＃39;青绿&＃39;:2,&＃39;乌黑&＃39;:3})
df[&＃39;根蒂&＃39;]&＃61;df[&＃39;根蒂&＃39;].map({&＃39;稍蜷&＃39;:1,&＃39;蜷缩&＃39;:2,&＃39;硬挺&＃39;:3})
df[&＃39;敲声&＃39;]&＃61;df[&＃39;敲声&＃39;].map({&＃39;清脆&＃39;:1,&＃39;浊响&＃39;:2,&＃39;沉闷&＃39;:3})
df[&＃39;纹理&＃39;]&＃61;df[&＃39;纹理&＃39;].map({&＃39;清晰&＃39;:1,&＃39;稍糊&＃39;:2,&＃39;模糊&＃39;:3})
df[&＃39;脐部&＃39;]&＃61;df[&＃39;脐部&＃39;].map({&＃39;平坦&＃39;:1,&＃39;稍凹&＃39;:2,&＃39;凹陷&＃39;:3})
df[&＃39;触感&＃39;] &＃61; np.where(df[&＃39;触感&＃39;]&＃61;&＃61;"硬滑",1,2)
df[&＃39;好瓜&＃39;] &＃61; np.where(df[&＃39;好瓜&＃39;]&＃61;&＃61;"是",1,0)
#由于西瓜数据集样本比较少&＃xff0c;所以不划分数据集&＃xff0c;将所有的西瓜数据用来训练模型
Xtrain &＃61; df.iloc[:,:-1]
Xtrain &＃61; np.array(Xtrain)
Ytrain &＃61; df.iloc[:,-1]
# 调用sklearn内置的决策树的库和画图工具
from sklearn import tree
import graphviz
# 采用ID3算法&＃xff0c;利用信息熵构建决策树模型
clf &＃61; tree.DecisionTreeClassifier(criterion&＃61;"entropy")clf &＃61; clf.fit(Xtrain,Ytrain)
##y_predict&＃61;clf.predict([[3,2,3,2,2,1]])
##print(y_predict)
##print(clf.feature_importances_)
# 绘制决策树的图形
feature_names &＃61; ["色泽","根蒂","敲声","纹理","脐部","触感"]
dot_data &＃61; tree.export_graphviz(clf,feature_names&＃61;feature_names,class_names&＃61;["好瓜","坏瓜"])
graph &＃61; graphviz.Source(dot_data)
graph.view()

西瓜集内容如下&＃xff1a; 

编号,色泽,根蒂,敲声,纹理,脐部,触感,好瓜
1,青绿,蜷缩,浊响,清晰,凹陷,硬滑,是
2,乌黑,蜷缩,沉闷,清晰,凹陷,硬滑,是
3,乌黑,蜷缩,浊响,清晰,凹陷,硬滑,是
4,青绿,蜷缩,沉闷,清晰,凹陷,硬滑,是
5,浅白,蜷缩,浊响,清晰,凹陷,硬滑,是
6,青绿,稍蜷,浊响,清晰,稍凹,软粘,是
7,乌黑,稍蜷,浊响,稍糊,稍凹,软粘,是
8,乌黑,稍蜷,浊响,清晰,稍凹,硬滑,是
9,乌黑,稍蜷,沉闷,稍糊,稍凹,硬滑,否
10,青绿,硬挺,清脆,清晰,平坦,软粘,否
11,浅白,硬挺,清脆,模糊,平坦,硬滑,否
12,浅白,蜷缩,浊响,模糊,平坦,软粘,否
13,青绿,稍蜷,浊响,稍糊,凹陷,硬滑,否
14,浅白,稍蜷,沉闷,稍糊,凹陷,硬滑,否
15,乌黑,稍蜷,浊响,清晰,稍凹,软粘,否
16,浅白,蜷缩,浊响,模糊,平坦,硬滑,否
17,青绿,蜷缩,沉闷,稍糊,稍凹,硬滑,否