52424aae5e284440efdb4aea30266a69553d1dde,umap/umap_.py,UMAP,transform,#UMAP#Any#,1665

Before Change

            indices = np.argsort(dmat)
            dists = np.sort(dmat)  // TODO: more efficient approach
            indices = indices[:, : self.n_neighbors]
            dists = dists[:, : self.n_neighbors]
        else:
            init = initialise_search(
                self._rp_forest,
                self._raw_data,

After Change

            dmat = pairwise_distances(
                X, self._raw_data, metric=self.metric, **self._metric_kwds
            )
            indices = np.argpartition(dmat,
                                      self._n_neighbors)[:, :self._n_neighbors]
            dmat_shortened = submatrix(dmat, indices, self._n_neighbors)
            indices_sorted = np.argsort(dmat_shortened)
            indices = submatrix(indices, indices_sorted, self._n_neighbors)
            dists = submatrix(dmat_shortened, indices_sorted,
                              self._n_neighbors)
        else:

Instances