From 663a98d6d8851da12320f68fba350405cd82e213 Mon Sep 17 00:00:00 2001 From: lucas_miranda <lucasmiranda42@gmail.com> Date: Mon, 20 Jul 2020 17:18:26 +0200 Subject: [PATCH] Early stopping now uses the overall loss function instead of the intercomponent overlap --- source/model_utils.py | 2 +- source/models.py | 2 +- 2 files changed, 2 insertions(+), 2 deletions(-) diff --git a/source/model_utils.py b/source/model_utils.py index fa953e8e..270c9098 100644 --- a/source/model_utils.py +++ b/source/model_utils.py @@ -275,7 +275,7 @@ class Entropy_regulariser(Layer): Identity layer that adds cluster weight entropy to the loss function """ - def __init__(self, weight=1., *args, **kwargs): + def __init__(self, weight=1.0, *args, **kwargs): self.weight = weight super(Entropy_regulariser, self).__init__(*args, **kwargs) diff --git a/source/models.py b/source/models.py index d3c0c9e0..a16c562b 100644 --- a/source/models.py +++ b/source/models.py @@ -173,7 +173,7 @@ class SEQ_2_SEQ_GMVAE: number_of_components=1, predictor=True, overlap_loss=False, - entropy_reg_weight=0., + entropy_reg_weight=1.0, ): self.input_shape = input_shape self.batch_size = batch_size -- GitLab