[人工智能]机器学习之模型评估

整体思路：先获取数据，把数据分为训练数据和测试数据，然后在不同k值的模型下分别训练和测试，得出不同k值情况下的模型预测准确性，最后把准确性可视化输出进行整体评估。

1.获取数据

from sklearn import datasets
iris=datasets.load_iris()
x=iris.data
y=iris.target

2.分离数据

from sklearn.model_selection import train_test_split
x_train,x_test,y_train,y_test=train_test_split(x,y,test_size=0.3)

test_size=0.3相当于测试数据的占比是0.3，而训练数据的占比是0.7，也可以通过下面代码验证

print(x_train.shape,x_test.shape,y_train.shape,y_test.shape)

(105, 4) (45, 4) (105,) (45,)

3.不同k值的模型下分别训练和测试，得出不同k值情况下的模型预测准确性

3.1创建一个1-25的列表

k_range=list(range(1,26))
print(k_range)

[1, 2, 3, 4, 5, 6, 7, 8, 9, 10, 11, 12, 13, 14, 15, 16, 17, 18, 19, 20, 21, 22, 23, 24, 25]

3.2 分别创建训练数据得分准确率空列表和测试数据得分准确率空列表，用于收遍历循环后的数据

train_score=[]
test_score=[]

3.3 导入建立模型模块和判断准确率模块

from sklearn.neighbors import KNeighborsClassifier

from sklearn.metrics import accuracy_score

3.4开始不同的k值遍历循环

for k in k_range:
    knn=KNeighborsClassifier(n_neighbors=k)
    knn.fit(x_train,y_train)
    y_train_predict=knn.predict(x_train)
    y_test_predict=knn.predict(x_test)
    a=accuracy_score(y_train,y_train_predict)
    b=accuracy_score(y_test,y_test_predict)
    train_score.append(a)
    test_score.append(b)

3.5因为一个模型的评估主要是针对测试数据进行预测的准确率的高低，所以只提取测试数据准确率

for k in k_range:
    print(k,test_score[k-1])

1 0.9555555555555556
2 0.9555555555555556
3 0.9555555555555556
4 0.9555555555555556
5 0.9777777777777777
6 0.9555555555555556
7 0.9777777777777777
8 0.9777777777777777
9 0.9777777777777777
10 0.9555555555555556
11 1.0
12 1.0
13 1.0
14 0.9777777777777777
15 1.0
16 1.0
17 1.0
18 0.9555555555555556
19 0.9777777777777777
20 0.9777777777777777
21 1.0
22 0.9555555555555556
23 0.9555555555555556
24 0.9333333333333333
25 0.9555555555555556

4.评估数据图表可视化

4.1导入绘图库模块

import matplotlib.pyplot as plt

4.2设置输出结果参数

%matplotlib inline

4.3开始输出训练准确率图标

plt.plot(k_range,train_score)
plt.xlabel('k_volue')
plt.ylabel('score')

?4.4开始输出测试数据准确率

plt.plot(k_range,test_score)
plt.xlabel('k_volue')
plt.ylabel('score')

?

从测试数据得分率看出不是k越小或者越大，准确率越高，几乎是随机的。

k值越小，模型越过于拟合，越复杂

《新程序员》：云原生和全面数字化实践

50位技术专家共同创作，文字、视频、音频交互阅读