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

Before Change

          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):

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[
        numpy.logical_not(m1)
      ]

      knn_erros = numpy.sum(m1)

      // Find labels of neighbors found in the training data.

Instances