Skip to content

Commit e285ca3

Browse files
committed
Add entropic_partial_gromov_wasserstein2 function
Exports and implements the entropic_partial_gromov_wasserstein2 function, providing a Julia wrapper for the corresponding POT library method.
1 parent e569c66 commit e285ca3

File tree

2 files changed

+5
-0
lines changed

2 files changed

+5
-0
lines changed

src/PythonOT.jl

Lines changed: 1 addition & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -15,6 +15,7 @@ export emd,
1515
empirical_sinkhorn_divergence,
1616
mm_unbalanced,
1717
entropic_partial_wasserstein,
18+
entropic_partial_gromov_wasserstein2,
1819
partial_wasserstein,
1920
partial_wasserstein2
2021

src/lib.jl

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -594,3 +594,7 @@ end
594594
function partial_wasserstein2(a, b, M; kwargs...)
595595
return pot.partial.partial_wasserstein2(a, b, M; kwargs...)
596596
end
597+
598+
function entropic_partial_gromov_wasserstein2(C1, C2, p, q; kwargs...)
599+
return pot.partial.entropic_partial_gromov_wasserstein2(C1, C2, p, q; kwargs...)
600+
end

0 commit comments

Comments
 (0)