expt: Trying Wasserstein

2021-10-06 18:25:08 +02:00
parent cec738d5cf
commit 192c6b8caf
4 changed files with 5 additions and 5 deletions
--- a/code/lib/lmdk_sel.py
+++ b/code/lib/lmdk_sel.py
@ -175,8 +175,8 @@ def get_opts_from_top_h(seq, lmdks):
        hist_tmp = np.copy(hist_cur)
        hist_tmp[i] += 1
        # Find difference from original
-        diff_cur = get_norm(hist, hist_tmp)  # Euclidean
-        # diff_cur = get_emd(hist, hist_tmp)  # Wasserstein
+        # diff_cur = get_norm(hist, hist_tmp)  # Euclidean
+        diff_cur = get_emd(hist, hist_tmp)  # Wasserstein
        # Remember if it is the best that you've seen
        if diff_cur < diff_min:
          diff_min = diff_cur