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