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