VAE model with KL annealing

ayushkarnawat / profit

Exploring evolutionary protein fitness landscapes

MIT License

1 stars 0 forks source link

VAE model with KL annealing #89

Closed ayushkarnawat closed 4 years ago

ayushkarnawat commented 4 years ago

What does this PR do?

Implements basic VAE model w/ KL annealing

This allows the model to learn a latent space embedding of the features. This can be used as a prior to sample new points to explore in the search space.