Skip to content

Instantly share code, notes, and snippets.

View rtavenar's full-sized avatar
💭
I may be slow to respond.

Romain Tavenard rtavenar

💭
I may be slow to respond.
View GitHub Profile
@rtavenar
rtavenar / parallel_cdist.py
Last active April 15, 2023 15:44
Use joblib to parallelize distance computations in cdist
import numpy
from joblib.parallel import Parallel, delayed
def cdist_generic(dist_fun, dataset1, dataset2=None, n_jobs=None verbose=0,
compute_diagonal=True, *args, **kwargs):
"""Compute cross-similarity matrix with joblib parallelization for a given
similarity function.
Parameters
----------