下你所需,载你所想!
汇集开发技术源码资料

机器学习K-近邻算法(KNN)

:1.215KB :1 :2022-01-07 17:53:29

部分简介

机器学习K-近邻算法(KNN)如果开发者对于本文件有需要的可以参考。
dataSetsize=dataSet.shape[0]#shape[0]返回dataset这个array的行数,代表有几个训练样本
diffMat=tile(inX,(dataSetsize,1))-dataSet
#tile()函数作用是将inX这个向量补成和datasize行1列的矩阵,这样就能和dataSet做减法了
sqDiffMat=diffMat**2#diffMat中的每个元素平方
sqDistances=sqDiffMat.sum(axis=1)#axis=1表示按行相加,axis=0表示按列相加
distances=sqDistances**0.5#开方
sortedDistIndicies=distances.argsort()#距离从小到大排序,返回的是排序后的下标组成的数组[2 3 1 0]

热门推荐

相关文章