deep-unimodal-ordinal

The code for the paper can be found here:

https://github.com/christopher-beckham/msc/tree/master/experiments/earth_mover

The Poisson and binomial extension functions/layers can be found:

https://github.com/christopher-beckham/msc/blob/master/experiments/earth_mover/architectures/resnet.py

For example, the binomial one is:

https://github.com/christopher-beckham/msc/blob/master/experiments/earth_mover/architectures/resnet.py#L189-L233

I haven't had time to do a nice clean-up, but let me know if you have any questions.