NumPy implementation of Poincaré Embeddings for Learning Hierarchical Representations (Facebook Research)
Switch branches/tags
Nothing to show
Clone or download
Fetching latest commit…
Cannot retrieve the latest commit at this time.
Permalink
Failed to load latest commit information.
plots
LICENSE
README.md
poincare.py
poincare_adam.py

README.md

poincare_embeddings

NumPy implementation of Poincaré Embeddings for Learning Hierarchical Representations (Facebook Research).
Link to paper
Link to blog

poincare.py contains the crude implementation, poincare_adam.py contains my experiments with adam.

Sample results:

Without adam, three levels deep:

Initial

https://github.com/nishnik/poincare_embeddings/blob/master/plots/3_init.png

Epoch 1

https://github.com/nishnik/poincare_embeddings/blob/master/plots/3_1.png

Epoch 181

https://github.com/nishnik/poincare_embeddings/blob/master/plots/3_181.png

With adam, three levels deep:

Initial

https://github.com/nishnik/poincare_embeddings/blob/master/plots/adam3_init.png

Epoch 1

https://github.com/nishnik/poincare_embeddings/blob/master/plots/adam3_1.png

Epoch 181

https://github.com/nishnik/poincare_embeddings/blob/master/plots/adam3_181.png

Without adam, four levels deep:

Initial

https://github.com/nishnik/poincare_embeddings/blob/master/plots/4_init.png

Epoch 1

https://github.com/nishnik/poincare_embeddings/blob/master/plots/4_1.png

Epoch 181

https://github.com/nishnik/poincare_embeddings/blob/master/plots/4_181.png

With adam, four levels deep:

Initial

https://github.com/nishnik/poincare_embeddings/blob/master/plots/adam4_init.png

Epoch 1

https://github.com/nishnik/poincare_embeddings/blob/master/plots/adam4_1.png

Epoch 181

https://github.com/nishnik/poincare_embeddings/blob/master/plots/adam4_181.png