Commit efaa08cb authored by lucas_miranda's avatar lucas_miranda
Browse files

test posterior with constant variance per component

parent d39233e7
......@@ -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=1
tf.multiply(z, tf.where(~tf.math.is_inf(K.log(z)), K.log(z), 0)), axis=0
)
# Adds metric that monitors dead neurons in the latent space
......
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