diff --git a/function.py b/function.py index 696ba1d..3735a8a 100644 --- a/function.py +++ b/function.py @@ -58,9 +58,9 @@ def classify0(inX, dataSet, labels,k=4): sortedDistIndicies=distances.argsort() classCount={} for i in range(k): - voteIlabel=labelssortedDistIndiciesi]] - classCount[voteIlabel]=classCount.get(voteIlabel,0)+1 - sortedClassCount = sorted(classCount,items(), key=lambda item: item[1l. reverse=True) + voteIlabel=labels[sortedDistIndicies[i]] + classCount[voteIlabel]=classCount.get(voteIlabel,0)+1 + sortedClassCount = sorted(classCount,items(), key=lambda item: item[1]. reverse=True) return sortedclassCount[0][0]