Not a member of Pastebin yet?
Sign Up,
it unlocks many cool features!
- 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
Add Comment
Please, Sign In to add comment