Poincaré Embeddings for Learning Hierarchical Representations using Distributed frameworks
Implementation of Poincaré Embeddings for Learning Hierarchical Representations using distributed frameworks such as TensorFlow, PyTorch and HOROVOD.
Installation
Simply clone this repository via
git clone https://github.com/sobalgi/DS222-Poincare-Embedding.gitcd DS222-Poincare-Embedding
Dependencies
Folders
For TensorFLow implementations
cd tensorflow
For PyTorch implementations
cd pytorch
For Numpy implementation
cd numpy
The steps to reproduce the results of the experiments are given in the respective readme files of the folders.
References
Original implementaion of Poincare Embeddings from Facebook AI Research. (with unresolved issues!!!)
FAIR code
Instructions to setup and use Horovod with PyTorch and TensorFlow.
HOROVOD