Skip to content

Commit 90877fa

Browse files
committed
Update lib.jl
Fix problems in doc generation
1 parent d62877b commit 90877fa

File tree

1 file changed

+4
-11
lines changed

1 file changed

+4
-11
lines changed

src/lib.jl

Lines changed: 4 additions & 11 deletions
Original file line numberDiff line numberDiff line change
@@ -555,20 +555,11 @@ The function considers the following problem:
555555
```math
556556
\\gamma = \\mathop{\\arg \\min}_\\gamma \\quad \\langle \\gamma,
557557
\\mathbf{M} \\rangle_F + \\mathrm{reg} \\cdot\\Omega(\\gamma)
558-
559-
s.t. \\gamma \\mathbf{1} &\\leq \\mathbf{a} \\\\
560-
\\gamma^T \\mathbf{1} &\\leq \\mathbf{b} \\\\
561-
\\gamma &\\geq 0 \\\\
562-
\\mathbf{1}^T \\gamma^T \\mathbf{1} = m
563-
&\\leq \\min\\{\\|\\mathbf{a}\\|_1, \\|\\mathbf{b}\\|_1\\} \\\\
564558
```
565559
566-
where :
567-
568-
- `M` is the metric cost matrix
569-
- ``\\Omega`` is the entropic regularization term, ``\\Omega=\\sum_{i,j} \\gamma_{i,j}\\log(\\gamma_{i,j})``
570560
- `a` and `b` are the sample weights
571-
- `m` is the amount of mass to be transported
561+
- `M` is the metric cost matrix
562+
- `reg` is a regularization term > 0
572563
573564
This function is a wrapper of the function
574565
[`entropic_partial_wasserstein`](https://pythonot.github.io/gen_modules/ot.partial.html#ot.partial.entropic_partial_wasserstein) in the
@@ -589,6 +580,8 @@ julia> round.(entropic_partial_wasserstein(a, b, M, 1, m=0.1), digits=2)
589580
2×2 Matrix{Float64}:
590581
0.06 0.02
591582
0.01 0.0
583+
```
584+
592585
"""
593586
function entropic_partial_wasserstein(a, b, M, reg; kwargs...)
594587
return pot.partial.entropic_partial_wasserstein(a, b, M, reg; kwargs...)

0 commit comments

Comments
 (0)