knn.py 文件源码

def __knn_sklearn(X, k, n_jobs=-1, verbose=False, **kwargs):

    nn = NearestNeighbors(n_neighbors=k+1, n_jobs=n_jobs,
                          algorithm='ball_tree', **kwargs)
    nn.fit(X)

    if verbose:
        print('Indexing done.')
    dist, ind = nn.kneighbors(X, k+1, return_distance=True)

    if verbose:
        print('Query done.')

    return dist[:,1:].astype(X.dtype), ind[:,1:]