Skip to content

Instantly share code, notes, and snippets.

@Guidosalimbeni
Created January 7, 2021 21:32
Show Gist options
  • Save Guidosalimbeni/174abdc207903182624a98932068359f to your computer and use it in GitHub Desktop.
Save Guidosalimbeni/174abdc207903182624a98932068359f to your computer and use it in GitHub Desktop.
def getNeighbors(trainingSet, testInstance, k):
distances = []
length = len(testInstance)-1
for x in range(len(trainingSet)):
dist = euclideanDistance(testInstance, trainingSet[x], length)
distances.append((trainingSet[x], dist))
distances.sort(key=operator.itemgetter(1))
neighbors = []
for x in range(k):
neighbors.append(distances[x][0])
return neighbors
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment