diff options
author | Vivien Seguy <vivienseguy@h40.57.229.10.1016746.vlan.kuins.net> | 2018-07-09 17:40:41 +0900 |
---|---|---|
committer | Vivien Seguy <vivienseguy@h40.57.229.10.1016746.vlan.kuins.net> | 2018-07-09 17:40:41 +0900 |
commit | af57d90c83c860db5a2160e79aea407ae379f7b0 (patch) | |
tree | 62ad6c10932f8e8d425d52fdd0eb0f0a91404b21 /ot/lp/__init__.py | |
parent | 08e5c0a91bc9b4afcd375109c08a14bcb0b1bd51 (diff) |
return log dict in free support barycenter function
Diffstat (limited to 'ot/lp/__init__.py')
-rw-r--r-- | ot/lp/__init__.py | 5 |
1 files changed, 4 insertions, 1 deletions
diff --git a/ot/lp/__init__.py b/ot/lp/__init__.py index 96bf6de..02cbd8c 100644 --- a/ot/lp/__init__.py +++ b/ot/lp/__init__.py @@ -278,7 +278,9 @@ def free_support_barycenter(measures_locations, measures_weights, X_init, b=None X = X_init + log_dict = {} displacement_square_norms = [] + displacement_square_norm = stopThr + 1. while ( displacement_square_norm > stopThr and iter_count < numItermax ): @@ -303,6 +305,7 @@ def free_support_barycenter(measures_locations, measures_weights, X_init, b=None iter_count += 1 if log: - return X, displacement_square_norms + log_dict['displacement_square_norms'] = displacement_square_norms + return X, log_dict else: return X
\ No newline at end of file |