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