Downloads · 30 days
0
clementchadebec/reproduced_wrapped_poincare_vae
reproduced_wrapped_poincare_vae is a machine learning model from clementchadebec. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for pythae. The card lists the license as apache-2.0.
This model was trained with pythae. It can be downloaded or reloaded using the method loadfromhfhub
Downloads · 30 days
0
Access
Public
Updated Sep 7, 2022
Repo size
15.3 MB
Likes
0
Public
Click a slice to open those files.
.pkl3.8 MB · 50%
From the Hugging Face model README
This model was trained with pythae. It can be downloaded or reloaded using the method load_from_hf_hub
>>> from pythae.models import AutoModel
>>> model = AutoModel.load_from_hf_hub(hf_hub_path="clementchadebec/reproduced_wrapped_poincare_vae")
This trained model reproduces the results of the official implementation of [1].
| Model | Dataset | Metric | Obtained value | Reference value |
|---|---|---|---|---|
| PoincareVAE | MNIST | NLL (500 IS) | 101.66 (0.00) | 101.47 (0.01) |
[1] Mathieu, E., Le Lan, C., Maddison, C. J., Tomioka, R., & Teh, Y. W. (2019). Continuous hierarchical representations with poincaré variational auto-encoders. Advances in neural information processing systems, 32.