68fe96add85bd842df23569dde490e1694d256fe,cleverhans/model_zoo/deep_k_nearest_neighbors/dknn.py,DkNNModel,find_train_knns,#DkNNModel#Any#,134

Before Change

      for i in range(data_activations_layer.shape[0]):
        query_res = self.query_objects[layer].find_k_nearest_neighbors(
            data_activations_layer[i], self.neighbors)
        try:
          knns_ind[layer][i, :] = query_res
        except:  // pylint: disable-msg=W0702
          knns_ind[layer][i, :len(query_res)] = query_res
          knn_errors += knns_ind[layer].shape[1] - len(query_res)

      // Find labels of neighbors found in the training data.
      knns_labels[layer] = np.zeros((nb_data, self.neighbors), dtype=np.int32)
      for data_id in range(nb_data):
        knns_labels[layer][data_id, :] = self.train_labels[knns_ind[layer][data_id]]

After Change

        data_activations_layer, self.neighbors
      )

      d1 = neighbor_index.reshape(-1)
      m1 = d1 == -1

      knns_ind[layer].reshape(-1)[
        numpy.logical_not(m1)
      ] = d1[

Instances