Replaced for loop with vectorised mapping on ClusterOverlap regularization layer

dc2d2e9a · lucas_miranda · 0d79da7f · dc2d2e9a · dc2d2e9a
Commit dc2d2e9a authored May 28, 2021 by lucas_miranda
--- a/deepof/model_utils.py
+++ b/deepof/model_utils.py
@@ -564,6 +564,6 @@ class ClusterOverlap(Layer):
            # minimize local entropy
            self.add_loss(self.loss_weight * tf.reduce_mean(neighbourhood_entropy))
            # maximize number of clusters
-            self.add_loss(-self.loss_weight * tf.reduce_mean(number_of_clusters))
+            # self.add_loss(-self.loss_weight * tf.reduce_mean(number_of_clusters))

        return encodings
--- a/supplementary_notebooks/deepof_explore_model_stability.ipynb
+++ b/supplementary_notebooks/deepof_explore_model_stability.ipynb