real_nvp / README.md
merve's picture
merve HF staff
Set `library_name` to `tf-keras`. (#1)
56374cf verified
|
raw
history blame
1.34 kB
metadata
library_name: tf-keras
tags:
  - generative

Model description

This model demonstrates real-valued non-volume preserving (real NVP) transformations, a tractable yet expressive approach to modeling high-dimensional data.

This model is used to map a simple distribution - which is easy to sample and whose density is simple to estimate - to a more complex one learned from the data. This kind of generative model is also known as "normalizing flow". The latent distribution we wish to map to in this example is Gaussian.

Training and evaluation data

This model is trained on a toy dataset, the make_moons from sklearn.datasets.

Requirements

This model requires

  • Tensorflow 2.9.1
  • Tensorflow probability 0.17.0

Loading this Model

Only the model weights are stored. To load this model, use load_model function from load_model.py.

Original Data Distribution and Latent Distribution

Data Distribution

Latent Distribution

Training hyperparameters

The following hyperparameters were used during training:

name learning_rate decay beta_1 beta_2 epsilon amsgrad training_precision
Adam 9.999999747378752e-05 0.0 0.8999999761581421 0.9990000128746033 1e-07 False float32