Commit dc2d2e9a authored by lucas_miranda's avatar lucas_miranda
Browse files

Replaced for loop with vectorised mapping on ClusterOverlap regularization layer

parent 0d79da7f
......@@ -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
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