Commit 663a98d6 authored by lucas_miranda's avatar lucas_miranda
Browse files

Early stopping now uses the overall loss function instead of the intercomponent overlap

parent 806428a3
...@@ -275,7 +275,7 @@ class Entropy_regulariser(Layer): ...@@ -275,7 +275,7 @@ class Entropy_regulariser(Layer):
Identity layer that adds cluster weight entropy to the loss function 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 self.weight = weight
super(Entropy_regulariser, self).__init__(*args, **kwargs) super(Entropy_regulariser, self).__init__(*args, **kwargs)
......
...@@ -173,7 +173,7 @@ class SEQ_2_SEQ_GMVAE: ...@@ -173,7 +173,7 @@ class SEQ_2_SEQ_GMVAE:
number_of_components=1, number_of_components=1,
predictor=True, predictor=True,
overlap_loss=False, overlap_loss=False,
entropy_reg_weight=0., entropy_reg_weight=1.0,
): ):
self.input_shape = input_shape self.input_shape = input_shape
self.batch_size = batch_size self.batch_size = batch_size
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment