diff options
| author | Mokhtar Z. Alaya <mokhtarzahdi.alaya@gmail.com> | 2020-01-07 19:00:50 +0100 |
|---|---|---|
| committer | GitHub <noreply@github.com> | 2020-01-07 19:00:50 +0100 |
| commit | cfe26140af85082197151d0dc50915c0bd71f602 (patch) | |
| tree | 03216656f88bfc1ffeafb03575e8e679aa028a0a /ot/bregman.py | |
| parent | 92b7075568207a468cc821cd6a21e130b9d89f96 (diff) | |
fix definitions complementary active sets Ic, Jc
Diffstat (limited to 'ot/bregman.py')
| -rw-r--r-- | ot/bregman.py | 4 |
1 files 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)] |