Commit b749ede1 authored by lucas_miranda's avatar lucas_miranda
Browse files

Early stopping now uses the overall loss function instead of the intercomponent overlap

parent f9dc866a
......@@ -286,7 +286,7 @@ class Entropy_regulariser(Layer):
def call(self, z, **kwargs):
entropy = K.sum(
tf.multiply(z, tf.where(~tf.math.is_inf(K.log(z)), K.log(z), 0)), axis=0
tf.multiply(z, tf.where(~tf.math.is_inf(K.log(z)), K.log(z), 0)), axis=1
)
# Adds metric that monitors dead neurons in the latent space
......
......@@ -197,7 +197,7 @@ class SEQ_2_SEQ_GMVAE:
if self.prior == "standard_normal":
init_means = far_away_uniform_initialiser(
[self.number_of_components, self.ENCODING], minval=0, maxval=15
[self.number_of_components, self.ENCODING], minval=0, maxval=5
)
self.prior = tfd.mixture.Mixture(
......
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