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

Epoch 1

Epoch 181

With adam, three levels deep:

Initial

Epoch 1

Epoch 181

Without adam, four levels deep:

Initial

Epoch 1

Epoch 181

With adam, four levels deep:

Initial

Epoch 1

Epoch 181