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:
),
)(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(
tfpl.IndependentNormal.params_size(
self.ENCODING * self.number_of_components
......@@ -417,13 +424,6 @@ class GMVAE:
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(
make_distribution_fn=lambda gauss: tfd.mixture.Mixture(
cat=tfd.categorical.Categorical(
......
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