Skip to content

Commit

Permalink
Unbalanced divergence (#170)
Browse files Browse the repository at this point in the history
* first stab at unbalanced sinkhorn divergence

* fixed point in log domain

* increment version number

* add docstrings for sinkhorn_divergence_unbalanced

* format

* clean emacs tmp files

* clean emacs tmp files

* missing math tag

* missing backslash

* add sinkhorn_divergence_unbalanced to docs

* increase epsilon for gpu sinkhorn tests

* format

* udpate examples/OneDimension

* add StatsPlots deps to examples/OneDimension

* change Julia LTS from 1.0 to 1.6

Co-authored-by: Stephen Zhang <[email protected]>
  • Loading branch information
zsteve and Stephen Zhang authored Aug 21, 2022
1 parent a82e3fb commit 9da044c
Show file tree
Hide file tree
Showing 15 changed files with 381 additions and 160 deletions.
2 changes: 1 addition & 1 deletion .github/workflows/CI.yml
Original file line number Diff line number Diff line change
Expand Up @@ -20,7 +20,7 @@ jobs:
strategy:
matrix:
version:
- '1.0'
- '1.6'
- '1'
- 'nightly'
os:
Expand Down
5 changes: 5 additions & 0 deletions .gitignore
Original file line number Diff line number Diff line change
Expand Up @@ -33,3 +33,8 @@ docs/src/examples/
# Files generated by Jupyter Notebooks
*.ipynb_checkpoints
*.ipynb

# emacs temp files
*~undo-tree~*
\#*\#
.\#*
2 changes: 1 addition & 1 deletion Project.toml
Original file line number Diff line number Diff line change
@@ -1,7 +1,7 @@
name = "OptimalTransport"
uuid = "7e02d93a-ae51-4f58-b602-d97af76e3b33"
authors = ["zsteve <[email protected]>"]
version = "0.3.19"
version = "0.3.20"

[deps]
ExactOptimalTransport = "24df6009-d856-477c-ac5c-91f668376b31"
Expand Down
1 change: 1 addition & 0 deletions docs/src/index.md
Original file line number Diff line number Diff line change
Expand Up @@ -54,6 +54,7 @@ sinkhorn_stabilized_epsscaling
```@docs
sinkhorn_unbalanced
sinkhorn_unbalanced2
sinkhorn_divergence_unbalanced
```

## Quadratically regularised optimal transport
Expand Down
Loading

0 comments on commit 9da044c

Please sign in to comment.