Commit 226b9801 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 cb1528bd
This source diff could not be displayed because it is too large. You can view the blob instead.
...@@ -722,9 +722,11 @@ class SEQ_2_SEQ_MMVAEP: ...@@ -722,9 +722,11 @@ class SEQ_2_SEQ_MMVAEP:
# Categorical prior on mixture of Gaussians # Categorical prior on mixture of Gaussians
categories = Dense(self.number_of_components, activation="softmax") categories = Dense(self.number_of_components, activation="softmax")
for _ in range(self.number_of_components): for i in range(self.number_of_components):
z_mean = Dense(self.ENCODING)(encoder) z_mean = Dense(self.ENCODING, name="{}_gaussian_mean".format(i+1))(encoder)
z_log_sigma = Dense(self.ENCODING)(encoder) z_log_sigma = Dense(self.ENCODING, name="{}_gaussian_sigma".format(i+1))(
encoder
)
# Define and control custom loss functions # Define and control custom loss functions
kl_warmup_callback = False kl_warmup_callback = False
......
Supports Markdown
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