eamonnbell/exponential_family_embeddings

Bernoulli Embeddings for Text

Bernoulli Embeddings for Text

This code implements Bernoulli Embeddings as described in this paper:

M. Rudolph, F. Ruiz, S. Mandt, D. Blei, Exponential Family Embeddings, Neural Information Processing Systems, 2016

Also, check out the NIPS 2016 Spotlight Video:

Getting Started

The code requires tensorflow, Edward and scikit learn.

Running Bernoulli Embeddings

To run, simply execute python main.py.