From cfe26140af85082197151d0dc50915c0bd71f602 Mon Sep 17 00:00:00 2001 From: "Mokhtar Z. Alaya" Date: Tue, 7 Jan 2020 19:00:50 +0100 Subject: fix definitions complementary active sets Ic, Jc --- ot/bregman.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/ot/bregman.py b/ot/bregman.py index 8cfea7e..c0634b1 100644 --- a/ot/bregman.py +++ b/ot/bregman.py @@ -1982,8 +1982,8 @@ def screenkhorn(a, b, M, reg, ns_budget=None, nt_budget=None, uniform=True, rest print('|I_active| = %s \t |J_active| = %s ' %(len(I), len(J))) # Ic, Jc: complementary of the active sets I and J - Ic = list(set(list(range(ns))) - set(I)) - Jc = list(set(list(range(nt))) - set(J)) + Ic = list(set(np.arange(ns)) - set(I)) + Jc = list(set(np.arange(nt)) - set(J)) K_IJ = K[np.ix_(I, J)] K_IcJ = K[np.ix_(Ic, J)] -- cgit v1.2.3