getNeighbors function

List getNeighbors(
  1. List train,
  2. List test_row,
  3. int num_neighbors
)

Implementation

List getNeighbors(List train, List test_row, int num_neighbors) {
  var distances = [];
  for (var i = 0; i < train.length - 1; i++) {
    var train_row = train[i];
    var dist = euclideanDistance(test_row, train_row);
    distances.add([train_row, dist]);
  }
  //sorting the while comparing the euclidean of the test_row and the train data
  var sorted_distance = sortDistanceList(distances);
  var neighbors = [];
  for (var i = 0; i < num_neighbors; i++) {
    neighbors.add(sorted_distance[i][0]);
  }
  return neighbors;
}