Skip to content

NumPy implementation of Poincaré Embeddings for Learning Hierarchical Representations (Facebook Research)

License

Notifications You must be signed in to change notification settings

nishnik/poincare_embeddings

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

5 Commits
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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

About

NumPy implementation of Poincaré Embeddings for Learning Hierarchical Representations (Facebook Research)

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published

Languages