当前位置:   article > 正文

KNN(K近邻算法)

knn

一、KNN基本知识 

  1. KNN算法 -------- 做分类(二分类、多分类)、也可以做回归
  2. ===================================
  3. KNN的三要素:
  4. K值的影响:
  5. 1. K值过小,可能会导致过拟合
  6. 2. K值过大,可能会导致欠拟合
  7. 距离的计算方式:
  8. 一般使用欧氏距离(欧几里得距离);
  9. 决策函数的选择:
  10. 在分类模型中,主要使用多数表决法或者加权多数表决法;在回归 模型中,主要使用平均值法或者加权平均值法。
  1. KNN算法知识脉络:
  2. -1. KNN算法是什么?KNN算法的基本原理是什么?
  3. -2. 损失函数是什么?
  4. -3. KNN伪代码怎么写的?基于Python实现KNN分类算法(多数投票的分类算法)
  5. -4. 你觉得有哪些因素可能影响KNN算法模型效果呢?你觉得KNN算法有什么缺点?怎么来解决?
  6. -5. 什么是KD树?KD树的创建策略?KD树的查找方式?
  1. 1. KNN算法是什么?KNN算法的基本原理是什么?
  2. KNN是有监督学习的K近邻的机器学习算法,K值得是最近的K个样本的意思;它的思想是 ‘近朱者赤近墨者黑’,若果空间中
  3. 某些样本具有相近的特征属性(样本距离比较近),我们可以认为它们的目标属性Y是相近的。我们可以用已有的最近K个样本
  4. 的目标属性来预测(分类:加权多票表决,回归:加权均值)待测样本的目标属性。
  5. 2. 损失函数是什么?
  6. KNN算法没有损失函数
  7. 3. KNN伪代码怎么写的?基于Python实现KNN分类算法(多数投票的分类算法)
  8. pass
  9. 4. 你觉得有哪些因素可能影响KNN算法模型效果呢?你觉得KNN算法有什么优缺点?怎么解决?
  10. 该算法适用于对样本容量比较大的类域进行自动分类。
  11. 4.1 你觉得有哪些因素可能影响KNN算法模型效果呢?
  12. ① KNN三要素的选择:
  13. K值的选择、距离计算方式、决策函数决策方式
  14. ② 样本不平衡问题(即有些类别的样本数量很多,而其它样本的数量很少),较少样本类别预测较差
  15. 4.2 你觉得KNN算法有什么优缺点?
  16. 4.2.1 KNN算法的优点:
  17. ① 思想简单,理论成熟,即可以做分类也可以做回归
  18. ② 可以天然解决多分类问题
  19. ③ 可以用非线性数据分类,即数据与数据存在交叉时效果也不错。
  20. ④ 训练时间复杂度为比SVM低
  21. ⑤ 准确度高,对异常点不敏感
  22. 4.2.1 你觉得KNN算法有什么优缺点?怎么解决?
  23. ① KNN算法的计算量很大(每次预测都需要计算所有的距离)
  24. 解决方案:KD树
  25. ② 样本不平衡问题(即有些类别的样本数量很多,而其它样本的数量很少),较少样本类别预测较差
  26. 解决方案:特征工程解决数据不均衡的方法(上下采样 ...)
  27. ③ 需要大量的内存;
  28. 5. 什么是KD树?KD树的创建策略?KD树的查找方式?
  29. 5.1 KD树是一个树状的数据结构,他是为了解决KNN算法样本数量过大时,计算距离最近K个样本速度过慢的问题。
  30. 5.2 KD Tree构建方式:
  31. KD树采用从m个样本的n维特征中,分别计算n个特征取值的方差,用方差最大 的第k维特征nk作为根节点。对于这个特征,选择取
  32. 值的中位数nkv作为样本的划分点,对于小于该值的样本划分到左子树,对于大于等于该值的样本划分到右子 树,对左右子树采用
  33. 同样的方式找方差最大的特征作为根节点,递归即可产生 KD树。
  34. 5.3 KD树的查找方式
  35. pass

 二、 KNN算法 

1、KNN算法的基本原理

 

 2、KNN三要素

3、KNN分类预测规则

4、KNN回归预测规则

 5、KNN算法实现方式

三、KD树 

  1. 1、KD树的目的:KD树是KNN算法为了快速求取K近邻样本的一种算法。
  2. 2、KD数的使用场景:用于样本数量比较大时,KKN算法快速计算距离最近K个样本的手段。
  3. 3、KD Tree构建方式:
  4. KD树采用从m个样本的n维特征中,分别计算n个特征取值的方差,用方差最大 的第k维特征nk作为根节点。对于这个特征,选择取
  5. 值的中位数nkv作为样本的划分点,对于小于该值的样本划分到左子树,对于大于等于该值的样本划分到右子 树,对左右子树采用
  6. 同样的方式找方差最大的特征作为根节点,递归即可产生 KD树。

 

  •          例: 

                

(2)KD tree查找最近邻 

 

  1. • 当我们生成KD树以后,就可以去预测测试集里面的样本目标点了。对于 一个目标点,我们首先在KD树里面找到包含目标点的叶子节
  2. 点。以目标 点为圆心,以目标点到叶子节点中样本实例的最短距离为半径,得到一 个超球体,最近邻的点一定在这个超球体内
  3. 部。然后返回叶子节点的父 节点,检查另一个子节点包含的超矩形体是否和超球体相交,如果相交 就到这个子节点寻找是否有
  4. 更加近的近邻,有的话就更新最近邻。如果不 相交那就简单了,我们直接返回父节点的父节点,在另一个子树继续搜 索最近邻。
  5. 当回溯到根节点时,算法结束,此时保存的最近邻节点就是 最终的最近邻。
  6. • 找到所属的叶子节点后,以目标点为圆心,以目标点到最近样本 点(一般为当前叶子节点中的其它训练数据或者刚刚经过的父节点)
  7. 为半径画圆,从最近样本点往根节点进行遍历,如果这个圆和分 割节点的分割线有交线,那么就考虑分割点的另外一个子树。如
  8. 果在遍历过程中,找到距离比刚开始的样本距离近的样本,那就 进行更新操作。
  9. • 一直迭代遍历到根节点上,结束循环找到最终的最小距离的样本。
  •    KD tree查找最近邻的图解

                                                

四、KNN的伪代码 与 sklearn 代码

 1、基于python的伪代码

  1. KNN伪代码:(将执行过程转换为代码的执行方式<代码+注释的形式>)
  2. -1. 简单来写
  3. def fit(self, X, Y, k=3):
  4. self.X_train = X
  5. self.Y_train = Y
  6. self.k = k
  7. def predict(self, X):
  8. result = []
  9. # 遍历所有的待预测样本,产生预测值,并将预测值保持到result临时集合中
  10. for x in X:
  11. # a. 计算当前样本x到训练数据集中的所有样本的距离
  12. # 并返回样本的下标和距离组成的集合,eg,[(距离1,样本1下标), (距离2, 样本2下标).....]
  13. all_distinces = self.calc_distinces(x)
  14. # b. 按照距离的远近对样本做一个排序的操作(升序)
  15. sort(all_distinces, key=lambda t: t[0])
  16. # c. 获取距离最近的K个样本对应的距离和下标组成的集合
  17. top_k_distinces = all_distinces[:self.k]
  18. # d. 从最近的样本距离、下标集合中获取样本下标
  19. top_k_indexs = list(map(lambda t:t[1], top_k_distinces))
  20. # e. 获取K个邻居样本对应的目标属性Y
  21. tmp_y = self.Y[top_k_indexs]
  22. # f. 统计一下临时集合tmp_y中各个类别取值出现的数量
  23. label_2_count_dict = self.calc_label_2_count(tmp_y)
  24. # g. 将字典转换为数组
  25. label_2_count_list = list(label_2_count_dict.items())
  26. # h. 对数据按照出现次数做一个排序(升序)
  27. sort(label_2_count_list, lambda t:t[1])
  28. # i. 获取出现次数最多的那个类别作为预测值
  29. tmp_label = label_2_count_list[-1][0]
  30. # j. 将预测值添加到集合中
  31. result.append(tmp_label)
  32. return result
  33. ------------------------------ 简化版伪代码 ---------------------------------------------
  34. def fit(self, X, Y, k=3):
  35. self.X_train = X
  36. self.Y_train = Y
  37. self.k = k
  38. def predict(self, X):
  39. result = []
  40. # 遍历所有的待预测样本,产生预测值,并将预测值保持到result临时集合中
  41. for x in X:
  42. # a. 获取和当前样本最相似的K个邻近样本的下标值
  43. top_k_indexs = self.fetch_nearst_neighbors(x)
  44. # b. 获取K个邻居样本对应的目标属性Y
  45. tmp_y = self.Y[top_k_indexs]
  46. # c. 计算K个样本中各个类别出现的概率,并获取概率最大的类别作为预测值
  47. tmp_label = self.fetch_max_proba_label(tmp_y)
  48. # d. 将预测值添加到集合中
  49. result.append(tmp_label)
  50. return result

 2、基于python的代码实现 KNN

  1. 数据:
  2. ======================= 基于python的KNN算法的伪代码 =====================================
  3. #_*_coding:utf-8_*_
  4. # 刘康
  5. import warnings,os,re
  6. import numpy as np
  7. import pandas as pd
  8. import matplotlib as mpl
  9. import matplotlib.pyplot as plt
  10. class Knn_code():
  11. def __init__(self):
  12. pass
  13. def fit(self,X,Y,k):
  14. '''
  15. 初始化样本
  16. :param X: 已知特征属性矩阵 X
  17. :param Y: 已知目标矩阵 Y
  18. :param k: KNN算法k值
  19. :return:
  20. '''
  21. self.X=X
  22. self.Y=Y
  23. self.k =k
  24. def calc_distance(self,once):
  25. '''
  26. 计算每一个待测样本与一直样本的距离的字典(key是已知样本的下标,value是与该样本的距离)
  27. '''
  28. distance_dict = {}
  29. for index in range(X.shape[0]):
  30. once_distance = np.sqrt(sum((X[index] - once)**2))
  31. distance_dict[index] = once_distance
  32. # print(distance_dict)
  33. return distance_dict
  34. def calc_label_counts_dict(self,label):
  35. '''
  36. 计算每个标签类型的数量,返回一个字典(key是每一个标签类型,value是该类型的数目)
  37. '''
  38. counts_dict = dict(pd.Series(label).value_counts())
  39. # print(counts,type(counts)) # Series 可以直接转化为字典
  40. return counts_dict
  41. def predict(self,X_predict):
  42. '''
  43. 对给定数据进行数据类型预测
  44. :param X_predict: 待预测 的样本的特征属性 矩阵X
  45. :return:
  46. '''
  47. result = []
  48. # 遍历所有的待预测样本,预测所有样本的类型,并保存带result列表中
  49. for once in X_predict:
  50. #计算被遍历单个样本与已知样本之间的距离字典(key是每个样本的索引,values与每个样本的距离)
  51. distance_dict = self.calc_distance(once)
  52. #将上述距离字典根据值进行升序排序,得到元组格式,即前k个是距离待测样本距离最近的k个样本
  53. distance_list = list(sorted(distance_dict.items(),key=lambda x:x[1]))
  54. # print(distance_list)
  55. #利用上诉元组键值对获取k个最近样本的索引列表
  56. index_list = list(map(lambda t:t[0],distance_list))[:self.k]
  57. # 根据上述索引取出对应样本的目标属性
  58. label_Y = self.Y[index_list]
  59. # print(label_Y)
  60. # 统计k个最近样本目标属性出现类别次数的字典
  61. Y_count_dict = self.calc_label_counts_dict(label_Y)
  62. # 将字典根据值的大小排序,返回列表
  63. Y_count_list = list(sorted(Y_count_dict.items(),key = lambda t:t[1]))
  64. # 获取上述列表最后一个元素中的类型
  65. Y_type = Y_count_list[-1][0]
  66. # print(Y_type)
  67. # 将改样本的类型预测保存
  68. result.append(Y_type)
  69. return np.array(result)
  70. if __name__ == '__main__':
  71. df_data = pd.read_csv('./iris.data',sep=',',header=None)
  72. print(df_data.info(),df_data.head())
  73. # 清洗数据
  74. df_data_label = df_data.iloc[:,-1]
  75. df_data_label_list = np.unique(df_data_label)
  76. # print(df_data_label_list)
  77. label_transform_dict = dict(zip(df_data_label_list,range(len(df_data_label_list))))
  78. # print(label_transform_dict)
  79. label_list = [label_transform_dict[i] for i in df_data_label]
  80. # print(label_list)
  81. # 提取特征属性X 目标属性Y
  82. X = np.array(df_data.iloc[:,:-1]) # 二维
  83. Y = np.array(label_list) # 一维
  84. knn = Knn_code()
  85. knn.fit(X,Y,10)
  86. # print(X)
  87. # print('='*100)
  88. # print(knn.Y)
  89. # 带预测样本
  90. X_test = np.array(df_data.iloc[:,:-1])
  91. # print(X_test)
  92. result_list = knn.predict(X_predict=X_test)
  93. print('样本真实值结果是:{}'.format(Y))
  94. print('样本预测的结果是:{}'.format(result_list))

3、基于sklearn库的KNN代码  

(1)ROC曲线与AUC值

  1. =============================== ROC曲线 与 AUC值 ======================================
  2. import matplotlib.pyplot as plt
  3. from sklearn import metrics
  4. if __name__ == '__main__':
  5. ------------------ 实际的类别y
  6. y_true = [0, 0, 1, 1, 0, 1, 0]
  7. ------------------ 决策函数值,或者是样本属于y=1的概率值
  8. y_scoreProd = [0.2, 0.45, 0.55, 0.68, 0.51, 0.9, 0.1]
  9. ------------------ 计算fpr, tpr以及阈值
  10. fpr, tpr, thresholds = metrics.roc_curve(y_true, y_scoreProd)
  11. print("FPR的值:{}".format(fpr))
  12. print("TPR的值:{}".format(tpr))
  13. print("阈值:{}".format(thresholds))
  14. auc = metrics.auc(x=fpr, y=tpr)
  15. print("面积:{}".format(auc))
  16. plt.plot([0, 0, 0, 1], [0, 1.0 / 3, 1.0, 1.0], 'r-o') ----------- 'r-o' 画出红线带点的图
  17. plt.show()

(2)sklearn库的KNN代码  

  1. ========================== 基于sklearn 库的 KNN算法模型 ===========================
  2. #_*_coding:utf-8_*_
  3. import os,re,warnings
  4. import numpy as np
  5. import pandas as pd
  6. import matplotlib as mpl
  7. import matplotlib.pyplot as plt
  8. import sklearn
  9. from sklearn.neighbors import KNeighborsClassifier
  10. from sklearn.preprocessing import LabelEncoder
  11. from sklearn.metrics import f1_score,accuracy_score,precision_score,recall_score,
  12. confusion_matrix,classification_report
  13. # 1、加载数据
  14. df_data = pd.read_csv('./iris.data',sep=',',header=None)
  15. # print(df_data.info(),df_data.head())
  16. # 2、清洗数据
  17. df_label_Y = df_data.iloc[:,-1]
  18. le = LabelEncoder()
  19. le.fit(df_label_Y)
  20. df_Y_trans = le.transform(df_label_Y) # 自动根据分类将标签进行编码 0,1,2,3 ......
  21. # print(df_Y_trans)
  22. # 3、提取特征属性X 目标属性Y
  23. X = df_data.iloc[:,:-1]
  24. Y = df_Y_trans
  25. # print(X,Y)
  26. # 4、分割训练集与测试集
  27. # 5、特征工程
  28. # 6、构建KNN模型
  29. '''
  30. class sklearn.neighbors.KNeighborsClassifier(n_neighbors=5, weights='uniform',
  31. algorithm='auto', leaf_size=30, p=2, metric='minkowski', metric_params=None, n_jobs=1, **kwargs)
  32. '''
  33. KNN_model = KNeighborsClassifier(n_neighbors=10,weights='distance',algorithm='kd_tree')
  34. # 7、训练模型
  35. KNN_model.fit(X,Y)
  36. # 8、模型评估、参数查看、可视化
  37. Y_predict = KNN_model.predict(X)
  38. print('KNN模型的正确率:{}'.format(KNN_model.score(X,Y)))
  39. # print('KNN模型的召回率:{}'.format(recall_score(Y,Y_predict))) # 注意默认的 二分类不能计算多分类的召回率
  40. print('KNN模型测试集的概率:{}'.format(KNN_model.predict_proba(X))) #返回的是一个 N * 类型数量列 的矩阵
  41. '''
  42. # kneighbors:从训练数据中获取样本X的邻近样本,获取n_neighbors参数多个样本,并且根据参数return_distance
  43. 决定是否返回距离值
  44. # 当前参数的情况下,返回的是一个二元组,二元组的第一个元素是一个集合,集合中存储的是邻近样本的距离;
  45. 第二个元素还是一个集合,存储的是邻近样本的下标
  46. kneighbors = algo.kneighbors(X=x_test, n_neighbors=9, return_distance=True)
  47. '''
  48. X_test = X.iloc[:3,:]
  49. print(X_test.shape)
  50. kneighbors = KNN_model.kneighbors(X=X_test,n_neighbors=3,return_distance=True)
  51. print('KNN模型测试集的最近的k个样本的距离、下标:{}'.format(kneighbors)) #
  52. print(kneighbors) # 返回的是一个索引值
  53. '''
  54. # kneighbors_graph, 对于X而言,从训练数据中获取距离最近的n_neighbors多个样本,将X和这些样本之间认为是可以连通的,
  55. 故将其值设置为1
  56. # mode给定值的设置方式,connectivity表示设置为1,distance表示设置为距离值
  57. # 返回对象表示的是X和训练数据中哪些样本是可以连通的
  58. # 列的数目是训练数据特征向量的数目,行的数目是x_test的样本数目
  59. kneighbors_graph(X=None, n_neighbors=None, mode='connectivity')
  60. '''
  61. kneighbors_graph = KNN_model.kneighbors_graph(X=X_test,n_neighbors=2,mode='distance')
  62. print('连接矩阵为:{}'.format(kneighbors_graph))
  63. '''
  64. 连接矩阵为: (0, 0) 0.0
  65. (0, 17) 0.09999999999999998
  66. (1, 1) 0.0 -------- 前三行是第一个样本的三个最近样本的链接情况,最前面是下标,后
  67. 面是距离
  68. (1, 45) 0.14142135623730986
  69. (2, 2) 0.0
  70. (2, 47) 0.14142135623730978
  71. '''
  72. # 混淆矩阵
  73. print('KNN模型的混淆矩阵为:{}'.format(confusion_matrix(Y,Y_predict)))
  74. # 9、模型部署持久化
  1. import numpy as np
  2. import pandas as pd
  3. import matplotlib.pyplot as plt
  4. from sklearn.preprocessing import LabelEncoder
  5. from sklearn.neighbors import KNeighborsClassifier
  6. from sklearn.metrics import classification_report, confusion_matrix
  7. from sklearn import metrics
  8. if __name__ == '__main__':
  9. # 1. 加载数据
  10. df = pd.read_csv('../datas/iris.data', header=None, names=['A', 'B', 'C', 'D', 'E'])
  11. # 删除一个类别的数据
  12. df = df[df.E != 'Iris-setosa']
  13. df.info()
  14. """
  15. 最终目的,是将X和Y转换为numpy数组的形式,并且值全部转换为数值型
  16. -1. 因为需要获取X和Y,所以这里第一步从df中提取对应的X、Y, 并将X和Y转换为numpy数组
  17. -2. 需要将Y中的字符串使用数字进行替换,eg:"Iris-setosa": 0, "Iris-versicolor": 1, "Iris-virginica": 2
  18. """
  19. # 获取X和Y
  20. columns = df.columns
  21. X = np.array(df[columns[:-1]])
  22. Y = np.array(df[columns[-1]])
  23. # 使用LabelEncoder对Y进行编码
  24. le = LabelEncoder()
  25. le.fit(Y)
  26. Y = le.transform(Y)
  27. print("Y中的类别:{}".format(le.classes_))
  28. # print("反转恢复:{}".format(le.inverse_transform([0, 1])))
  29. # 2. 获取X的最后两列数据作为原始的特征属性X
  30. # 按照索引获取所有行的最后两列的数据
  31. X = X[:, -2:]
  32. # print(X)
  33. # 3. 构建模型
  34. algo = KNeighborsClassifier(n_neighbors=9)
  35. # 4. 模型训练
  36. algo.fit(X, Y)
  37. # 5. 看模型效果
  38. Y_predict = algo.predict(X)
  39. print("准确率:{}".format(algo.score(X, Y)))
  40. print("训练数据上的分类报告:\n{}".format(classification_report(Y, Y_predict)))
  41. # 更改阈值(我希望将所有有病的数据全部预测正确)
  42. threshold = 0.77777778
  43. # 获取所有的预测概率
  44. test_predict_prob = algo.predict_proba(X)
  45. # 获取下标为1的那一列概率
  46. test_predict_prob = test_predict_prob[:, 1]
  47. # 做一个布尔索引,将所有值大于等于阈值的区域设置为4.0,小于阈值的区域设置为2.0
  48. test_predict_prob[test_predict_prob >= threshold] = 1
  49. test_predict_prob[test_predict_prob < threshold] = 0
  50. print("测试数据上的混淆矩阵:\n{}".format(confusion_matrix(Y, Y_predict)))
  51. print("更改阈值后测试数据上的混淆矩阵:\n{}".format(confusion_matrix(Y, test_predict_prob)))
  52. -------------------------- 多分类的情况不适合以下方法更改阈值、计算roc曲线,这里只谈论二分类更改阈值 ---------
  53. # 需要计算一下AUC和ROC的值
  54. # fpr、tpr和阈值 --> 需要实际值y和决策函数或者预测概率
  55. test_predict_prob = algo.predict_proba(X)
  56. print(test_predict_prob)
  57. # 将类别y=1当做正例则获取预测为类别y=1的概率作为auc计算过程中的阈值列表对象
  58. test_predict_prob = test_predict_prob[:, 1]
  59. # 计算fpr、tpr和阈值
  60. fpr, tpr, thresholds = metrics.roc_curve(Y, test_predict_prob)
  61. # 计算AUC面积
  62. auc = metrics.auc(x=fpr, y=tpr)
  63. print("AUC面积:{}".format(auc))
  64. print(thresholds)
  65. plt.plot(fpr, tpr, 'r-o')
  66. plt.show()
  1. ============================ KNN多分类 ROC曲线与AUC值 ==================================
  2. import numpy as np
  3. import pandas as pd
  4. import matplotlib.pyplot as plt
  5. from sklearn.preprocessing import LabelEncoder
  6. from sklearn.neighbors import KNeighborsClassifier
  7. from sklearn.metrics import classification_report, confusion_matrix
  8. from sklearn import metrics
  9. if __name__ == '__main__':
  10. # 1. 加载数据
  11. df = pd.read_csv('../datas/iris.data', header=None, names=['A', 'B', 'C', 'D', 'E'])
  12. df.info()
  13. """
  14. 最终目的,是将X和Y转换为numpy数组的形式,并且值全部转换为数值型
  15. -1. 因为需要获取X和Y,所以这里第一步从df中提取对应的X、Y, 并将X和Y转换为numpy数组
  16. -2. 需要将Y中的字符串使用数字进行替换,eg:"Iris-setosa": 0, "Iris-versicolor": 1, "Iris-virginica": 2
  17. """
  18. # 获取X和Y
  19. columns = df.columns
  20. X = np.array(df[columns[:-1]])
  21. Y = np.array(df[columns[-1]])
  22. # 使用LabelEncoder对Y进行编码
  23. le = LabelEncoder()
  24. le.fit(Y)
  25. Y = le.transform(Y)
  26. print("Y中的类别:{}".format(le.classes_))
  27. # print("反转恢复:{}".format(le.inverse_transform([0, 1])))
  28. # 2. 获取X的最后两列数据作为原始的特征属性X
  29. # 按照索引获取所有行的最后两列的数据
  30. X = X[:, -2:]
  31. # print(X)
  32. # 3. 构建模型
  33. algo = KNeighborsClassifier(n_neighbors=9)
  34. # 4. 模型训练
  35. algo.fit(X, Y)
  36. # 5. 看模型效果
  37. Y_predict = algo.predict(X)
  38. print("准确率:{}".format(algo.score(X, Y)))
  39. print("训练数据上的分类报告:\n{}".format(classification_report(Y, Y_predict)))
  40. # 更改阈值(我希望将所有有病的数据全部预测正确)
  41. threshold = 0.77777778
  42. # 获取所有的预测概率
  43. test_predict_prob = algo.predict_proba(X)
  44. # 获取下标为1的那一列概率
  45. test_predict_prob = test_predict_prob[:, 1]
  46. # 做一个布尔索引,将所有值大于等于阈值的区域设置为4.0,小于阈值的区域设置为2.0
  47. test_predict_prob[test_predict_prob >= threshold] = 1
  48. test_predict_prob[test_predict_prob < threshold] = 0
  49. print("测试数据上的混淆矩阵:\n{}".format(confusion_matrix(Y, Y_predict)))
  50. print("更改阈值后测试数据上的混淆矩阵:\n{}".format(confusion_matrix(Y, test_predict_prob)))
  51. # 需要计算一下AUC和ROC的值
  52. # 分别计算AUC的值
  53. test_predict_prob = algo.predict_proba(X)
  54. print(test_predict_prob)
  55. print(Y)
  56. # TODO: 三段代码合并,也就是说要求不允许直接出现0 1 2这三个数字
  57. # 对于第一个类别的AUC值的计算(y=0)
  58. # 构建计算fpr和tpr的时候的实际值,将原始数据中所有类别为0的设置为1,其他类别设置为0
  59. y1_true = (Y == 0).astype(np.int)
  60. y1_score = test_predict_prob[:, 0]
  61. fpr1, tpr1, _ = metrics.roc_curve(y1_true, y1_score)
  62. auc1 = metrics.auc(fpr1, tpr1)
  63. print("类别1的AUC值:{}".format(auc1))
  64. # 对于第二个类别的AUC值的计算(y=1)
  65. # 构建计算fpr和tpr的时候的实际值,将原始数据中所有类别为1的设置为1,其他类别设置为0
  66. y2_true = (Y == 1).astype(np.int)
  67. y2_score = test_predict_prob[:, 1]
  68. fpr2, tpr2, _ = metrics.roc_curve(y2_true, y2_score)
  69. auc2 = metrics.auc(fpr2, tpr2)
  70. print("类别2的AUC值:{}".format(auc2))
  71. # 对于第三个类别的AUC值的计算(y=2)
  72. # 构建计算fpr和tpr的时候的实际值,将原始数据中所有类别为2的设置为1,其他类别设置为0
  73. y3_true = (Y == 2).astype(np.int)
  74. y3_score = test_predict_prob[:, 2]
  75. fpr3, tpr3, _ = metrics.roc_curve(y3_true, y3_score)
  76. auc3 = metrics.auc(fpr3, tpr3)
  77. plt.plot(fpr1, tpr1, 'r-o')
  78. plt.plot(fpr2, tpr2, 'g-o')
  79. plt.plot(fpr3, tpr3, 'b-o')
  80. plt.show()
  81. ================================= 一起计算 ROC与AUC值 ===================================
  82. knn = KNeighborsClassifier(n_neighbors=3)
  83. knn.fit(X_train, Y_train)
  84. # b. 模型效果输出
  85. ## 将正确的数据转换为矩阵形式
  86. y_test_hot = label_binarize(Y_test,classes=(1,2,3)) ---------- 该方法的重点
  87. ## 得到预测属于某个类别的概率值
  88. knn_y_score = knn.predict_proba(X_test)
  89. ## 计算roc的值
  90. knn_fpr, knn_tpr, knn_threasholds = metrics.roc_curve(y_test_hot.ravel(),knn_y_score.ravel())
  91. ## 计算auc的值
  92. knn_auc = metrics.auc(knn_fpr, knn_tpr)
  93. print ("KNN算法准确率:", knn.score(X_train, Y_train))
  94. print ("KNN算法AUC值:", knn_auc)
  95. # c. 模型预测
  96. knn_y_predict = knn.predict(X_test)

四、KNN算法相关API

sklearn.neighbors
  1. neighbors.NearestNeighbors([n_neighbors,...]) 用于实现邻居搜索的无监督学习者。
  2. neighbors.KNeighborsClassifier([...]) 实现k近邻投票的分类器。
  3. neighbors.RadiusNeighborsClassifier([...]) 在给定半径内的邻居之间实施投票的分类器
  4. neighbors.KNeighborsRegressor([n_neighbors,...]) 基于k-最近邻居的回归。
  5. neighbors.RadiusNeighborsRegressor([radius,...]) 基于固定半径内的邻居的回归。
  6. neighbors.NearestCentroid([公制,......]) 最近的质心分类器。
  7. neighbors.BallTree BallTree用于快速广义的N点问题
  8. neighbors.KDTree KDTree用于快速广义N点问题
  9. neighbors.LSHForest([n_estimators,radius,...]) 使用LSH林执行近似最近邻搜索。
  10. neighbors.DistanceMetric DistanceMetric类
  11. neighbors.KernelDensity([带宽,......]) 核密度估计
  12. neighbors.kneighbors_graph(X,n_neighbors [,...]) 计算X中点的k-邻居的(加权)图
  13. neighbors.radius_neighbors_graph(X,半径) 计算X中点的邻居(加权)图

1、 KNN

(1)KNN分类

  1. class sklearn.neighbors.KNeighborsClassifier(n_neighbors=5, weights='uniform',
  2. algorithm='auto', leaf_size=30, p=2, metric='minkowski', metric_params=None, n_jobs=1,
  3. **kwargs)
  4. '''
  5. n_neighbors=5, ------- K 值
  6. weights='uniform' ------- 权重的计算方式
  7. 'uniform':均匀的权重。每个社区的所有积分均等。
  8. 'distance':权重点距离的倒数。在这种情况下,查询点的较近邻居将比远离的邻居具有更大的影响力。
  9. [callable]:一个用户定义的函数,它接受一个距离数组,并返回一个包含权重的相同形状的数组。
  10. algorithm : {‘auto’, ‘ball_tree’, ‘kd_tree’, ‘brute’}, optional ----- 默认即可
  11. leaf_size=30
  12. p=2, metric='minkowski' -------- 默认即可
  13. '''
  1. 属性:
  2. NO
  3. 方法:
  4. 1、fit(X, y) Fit the model using X as training data and y as target values
  5. 2get_params([deep]) Get parameters for this estimator.
  6. 3、kneighbors([X, n_neighbors, return_distance]) 找到一个点的K邻居。
  7. '''
  8. # kneighbors:从训练数据中获取样本X的邻近样本,获取n_neighbors参数多个样本,并且根据参数return_distance
  9. 决定是否返回距离值
  10. # 当前参数的情况下,返回的是一个二元组,二元组的第一个元素是一个集合,集合中存储的是邻近样本的距离;
  11. 第二个元素还是一个集合,存储的是邻近样本的下标
  12. kneighbors = algo.kneighbors(X=x_test, n_neighbors=9, return_distance=True)
  13. '''
  14. 4、kneighbors_graph = KNN_model.kneighbors_graph(X=X_test,n_neighbors=2,mode='distance')
  15. print('连接矩
  16. 阵为:{}'.format(kneighbors_graph)) 计算X中点的k-邻居的(加权)图
  17. '''
  18. # kneighbors_graph, 对于X而言,从训练数据中获取距离最近的n_neighbors多个样本,将X和这些样本之间认为是
  19. 可以连通的,故将其值设置为1
  20. # mode给定值的设置方式,connectivity表示设置为1,distance表示设置为距离值
  21. # 返回对象表示的是X和训练数据中哪些样本是可以连通的
  22. # 列的数目是训练数据特征向量的数目,行的数目是x_test的样本数目
  23. kneighbors_graph(X=None, n_neighbors=None, mode='connectivity')
  24. 例:
  25. 连接矩阵为: (0, 0) 0.0
  26. (0, 17) 0.09999999999999998
  27. (1, 1) 0.0 -------- 前三行是第一个样本的三个最近样本的链接情况,最前面是下标,后
  28. 面是距离
  29. (1, 45) 0.14142135623730986
  30. (2, 2) 0.0
  31. (2, 47) 0.14142135623730978
  32. '''
  33. 5、predict(X) Predict the class labels for the provided data
  34. 6、predict_proba(X) Return probability estimates for the test data X.
  35. 7、score(X, y[, sample_weight]) Returns the mean accuracy on the given test data and labels.
  36. 8set_params(\*\*params) Set the parameters of this estimator.

 

(2)KNN回归

  1. class sklearn.neighbors.KNeighborsRegressor(n_neighbors=5, weights='uniform',
  2. algorithm='auto', leaf_size=30, p=2, metric='minkowski', metric_params=None, n_jobs=1,
  3. **kwargs)

 2、KDtree

  1. sklearn.neighbors.KDTree
  2. KDTree(X, leaf_size=40, metric=’minkowski’, **kwargs)
  3. 方法:
  4. query(X, k=1, return_distance=True, dualtree=False, breadth_first=False) ---- 返回是索引与距离
  1. from sklearn.neighbors import KDTree
  2. from sklearn.datasets import make_blobs
  3. data,label = make_blobs(n_samples=100,n_features=3,centers=2,shuffle=True,random_state=28)
  4. # print(data,label)
  5. kdtree = KDTree(data,leaf_size=30)
  6. X_test = [[-8.48575193e+00 ,-5.60369601e+00 , 8.30169601e-01]]
  7. tmp = kdtree.query(X_test,k=5,return_distance=True)
  8. print(tmp)
  9. 结果:
  10. (array([[11.09321128, 11.11411761, 11.28784938, 11.29144955, 11.4498738 ]]), array([[34, 19, 14, 72,
  11. 75]], dtype=int64))

 

 

 

 

 

 

 

 

 

 

 

 

声明:本文内容由网友自发贡献,不代表【wpsshop博客】立场,版权归原作者所有,本站不承担相应法律责任。如您发现有侵权的内容,请联系我们。转载请注明出处:https://www.wpsshop.cn/w/我家自动化/article/detail/160066
推荐阅读
相关标签
  

闽ICP备14008679号