-
Notifications
You must be signed in to change notification settings - Fork 508
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
[MRG] MM algorithms for UOT #362
Conversation
Codecov Report
@@ Coverage Diff @@
## master #362 +/- ##
==========================================
- Coverage 93.78% 93.78% -0.01%
==========================================
Files 23 23
Lines 5845 5902 +57
==========================================
+ Hits 5482 5535 +53
- Misses 363 367 +4 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Thank you @lchapel this is a very nice PR.
could you pease add a few more tests to check for special cases ?
also I'm not sure why the doc is not building but please check locally that the new function give a proper documentation.
|
||
while (err > stopThr and cpt < numItermax): | ||
Kprev = K | ||
K = K * q1 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
What is all this? was the implementation false? why is there so much more steps?
ot/unbalanced.py
Outdated
if G0 is None: | ||
G = a[:, None] * b[None, :] | ||
else: | ||
G = G0 |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Please also test with G0 not given and with a and b =[]
log['err'].append(err) | ||
log['G'].append(G) | ||
if verbose: | ||
print('{:5d}|{:8e}|'.format(i, err)) |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
test with verbose
verbose=verbose, log=True) | ||
|
||
if log: | ||
return log_mm['cost'], log_mm |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
also test with log
Types of changes
Motivation and context / Related issue
How has this been tested (if it applies)
pep8 + doctest
PR checklist