From efaa08cbf5fd56806a7c86e4eca1c985ff73e6d2 Mon Sep 17 00:00:00 2001 From: lucas_miranda <lucasmiranda42@gmail.com> Date: Mon, 20 Jul 2020 22:34:32 +0200 Subject: [PATCH] test posterior with constant variance per component --- source/model_utils.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/source/model_utils.py b/source/model_utils.py index 72b354ac..270c9098 100644 --- a/source/model_utils.py +++ b/source/model_utils.py @@ -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 -- GitLab