42fcef3e6d5c79f0d9772e8b9f63338c3e5f715e,pynndescent/sparse_nndescent.py,,sparse_init_from_random,#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#Any#,308

Before Change

        to_data = query_data[query_indptr[i] : query_indptr[i + 1]]

        for j in range(indices.shape[0]):
            if indices[j] < 0:
                continue

            from_inds = inds[indptr[indices[j]] : indptr[indices[j] + 1]]
            from_data = data[indptr[indices[j]] : indptr[indices[j] + 1]]

            d = sparse_dist(from_inds, from_data, to_inds, to_data, *dist_args)

After Change

            to_data = query_data[query_indptr[i] : query_indptr[i + 1]]

            for j in range(np.sum(heap[0, i] == -1)):
                idx = np.abs(tau_rand_int(rng_state)) % (indptr.shape[0] - 1)

                from_inds = inds[indptr[idx] : indptr[idx + 1]]
                from_data = data[indptr[idx] : indptr[idx + 1]]

Instances