tumourkit.postprocessing.compare_barcodes.compute_wasserstein_distance

tumourkit.postprocessing.compare_barcodes.compute_wasserstein_distance(dgm1: ndarray, dgm2: ndarray, q: int) → float

Computes the q-Wasserstein distance between two persistence diagrams.

Parameters:

• dgm1 (np.ndarray) – Persistence diagram 1 as numpy array of (birth, death) pairs.

• dgm2 (np.ndarray) – Persistence diagram 2 as numpy array of (birth, death) pairs.

• q (int) – Order of the Wasserstein distance. None if bottleneck.

Returns:

Computed Wasserstein distance.

Return type:

float