From e46d76a7b9b1b17d1b4e3d1fa024b37e91de4190 Mon Sep 17 00:00:00 2001
From: lucas_miranda <lucasmiranda42@gmail.com>
Date: Mon, 6 Jul 2020 18:25:33 +0200
Subject: [PATCH] Implemented shuffle parameter in preprocessing; shuffled
validation data in model_training.py
---
source/model_utils.py | 8 ++++----
1 file changed, 4 insertions(+), 4 deletions(-)
diff --git a/source/model_utils.py b/source/model_utils.py
index ac45f847..a2e824ce 100644
--- a/source/model_utils.py
+++ b/source/model_utils.py
@@ -198,11 +198,11 @@ class Gaussian_mixture_overlap(Layer):
locs = (target[..., : self.lat_dims, k],)
scales = tf.keras.activations.softplus(target[..., self.lat_dims :, k])
- dists.append(tfd.BatchReshape(tfd.MultivariateNormalDiag(locs, scales), [-1]))
+ dists.append(
+ tfd.BatchReshape(tfd.MultivariateNormalDiag(locs, scales), [-1])
+ )
- print(dists)
dists = [tf.transpose(gauss.sample(self.samples), [1, 0, 2]) for gauss in dists]
- print(dists)
if self.metric == "mmd":
@@ -215,7 +215,7 @@ class Gaussian_mixture_overlap(Layer):
dtype=tf.float32,
)
)
- print(intercomponent_mmd)
+
self.add_metric(
intercomponent_mmd, aggregation="mean", name="intercomponent_mmd"
)
--
GitLab