pyFM.spectral.nn_utils

Functions

knn_query(X, Y[, k, return_distance, n_jobs])

Query nearest neighbors.
pyFM.spectral.nn_utils.knn_query(X, Y, k=1, return_distance=False, n_jobs=1)

Query nearest neighbors.

Parameters:

  • X (np.ndarray) – (n1,p) first collection

  • Y (np.ndarray) – (n2,p) second collection

  • k (int) – number of neighbors to look for

  • return_distance – whether to return the nearest neighbor distance

  • n_jobs – number of parallel jobs. Set to -1 to use all processes

Returns:

  • dists (np.ndarray) – (n2,k) or (n2,) if k=1 - ONLY if return_distance is False. Nearest neighbor distance.

  • matches (np.ndarray) – (n2,k) or (n2,) if k=1 - nearest neighbor