Commit a9e69c12 authored by lucas_miranda's avatar lucas_miranda
Browse files

Removed outdated non-variational autoencoder model

parent 6d1ad201
Pipeline #100397 canceled with stages
in 15 minutes and 54 seconds
...@@ -390,6 +390,13 @@ class GMVAE: ...@@ -390,6 +390,13 @@ class GMVAE:
), ),
)(encoder) )(encoder)
if self.overlap_loss:
z_cat = deepof.model_utils.Cluster_overlap(
self.ENCODING,
self.number_of_components,
loss_weight=self.overlap_loss,
)(z_cat)
z_gauss_mean = Dense( z_gauss_mean = Dense(
tfpl.IndependentNormal.params_size( tfpl.IndependentNormal.params_size(
self.ENCODING * self.number_of_components self.ENCODING * self.number_of_components
...@@ -417,13 +424,6 @@ class GMVAE: ...@@ -417,13 +424,6 @@ class GMVAE:
z_gauss = Reshape([2 * self.ENCODING, self.number_of_components])(z_gauss) z_gauss = Reshape([2 * self.ENCODING, self.number_of_components])(z_gauss)
if self.overlap_loss:
z_gauss = deepof.model_utils.Cluster_overlap(
self.ENCODING,
self.number_of_components,
loss_weight=self.overlap_loss,
)(z_gauss)
z = tfpl.DistributionLambda( z = tfpl.DistributionLambda(
make_distribution_fn=lambda gauss: tfd.mixture.Mixture( make_distribution_fn=lambda gauss: tfd.mixture.Mixture(
cat=tfd.categorical.Categorical( cat=tfd.categorical.Categorical(
......
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