机器学习之决策树算法（四）考试训练

题目要求

（一）题目要求
1.按要求完成下面的各项需求。
利用决策树实现分类问题:
加载西瓜数据
调用决策树分类模型，并训练模型
计算出深度=2和5时的准确率

数据如下

0.697,0.46,1
0.774,0.376,1
0.634,0.264,1
0.608,0.318,1
0.556,0.215,1
0.403,0.237,1
0.481,0.149,1
0.437,0.211,1
0.666,0.091,0
0.243,0.267,0
0.245,0.057,0
0.343,0.099,0
0.639,0.161,0
0.657,0.198,0
0.36,0.37,0
0.593,0.042,0
0.719,0.103,0

代码如下

# 1.加载西瓜数据(20分)
# 2.调用决策树分类模型，并训练模型(40分)
# 3.计算出深度=2和5时的准确率(40分)
import numpy as np
from sklearn.tree import DecisionTreeClassifier
# 读取数据
data = np.loadtxt(r'xigua.txt',delimiter=',')
# 提取数据
X = data[:,:-1]
y = data[:,-1]
# 特征缩放
X -= np.mean(X,axis=0)
X /= np.std(X,axis=0,ddof=1)

# 调用模型
model2 = DecisionTreeClassifier(max_depth=2)
model5 = DecisionTreeClassifier(max_depth=5)
# 训练模型
model2.fit(X,y)
model5.fit(X,y)

# 输出准确率
print('model2的准确率',model2.score(X,y))
print('model5的准确率',model5.score(X,y))

可视化如下