Commit b1862ad9 authored by lucas_miranda's avatar lucas_miranda
Browse files

Instanciated a mean and a variance for each component, and the categorical...

Instanciated a mean and a variance for each component, and the categorical prior as a Dense layer with a softmax activation
parent ce80ff28
This diff is collapsed.
......@@ -852,11 +852,10 @@ class SEQ_2_SEQ_MMVAEP:
# TODO:
# - Models as promer tf.keras.Model subclasses
# - Gaussian Mixture + Categorical priors -> Deep Clustering
# - MCMC sampling (n>1)
#
# TODO (in the non-immediate future):
# - MCMC sampling (n>1)
# - free bits paper
# - Attention mechanism for encoder / decoder (does it make sense?)
# - Transformer encoder/decoder (does it make sense?)
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment