alexeyev/sinkhornlayer-keras

Permutation learning Sinkhorn layer implementation in keras

SinkhornLayer-keras

Sinkhorn layer in keras, inspired by the reference implementation of numbers sorting task solution (please see the 'Learning Latent Permutations with Gumbel-Sinkhorn Networks' paper by Gonzalo Mena, David Belanger, Scott Linderman, Jasper Snoek).

Requirements:

• Python3
• keras, scipy, numpy

Bug reports, comments and enhancements are welcome.

Citation

Not neccessary, but is greatly appreciated.

@misc{Alekseev2018sinkhornkeras,
  author = {Alekseev~A.M.},
  title = {Sinkhorn layer, Keras implementation.},
  year = {2018},
  publisher = {GitHub},
  journal = {GitHub repository},
  howpublished = {\url{https://github.com/alexeyev/SinkhornLayer-keras}},
  commit = {the latest commit of the codebase you have used}
}