Poincaré Embeddings for Learning Hierarchical Representations

This is the NumPy and PyTorch implementation. Link to paper

The difference of results between implementations is due to the different sampling methods.

Results:

NumPy implementation

PyTorch implementation

TensorFlow implementation