poincare_embedding icon indicating copy to clipboard operation
poincare_embedding copied to clipboard

Poincaré Embedding

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

https://github.com/qiangsiwei/poincare_embedding/blob/master/data/mammal_numpy.png

PyTorch implementation

https://github.com/qiangsiwei/poincare_embedding/blob/master/data/mammal_torch.png

TensorFlow implementation

https://github.com/qiangsiwei/poincare_embedding/blob/master/data/mammal_tensor.png