Commit 7c5899dd authored by lucas_miranda's avatar lucas_miranda
Browse files

Added number of populated clusters as a training metric

parent 9adbeaf5
Pipeline #100368 passed with stages
in 23 minutes and 5 seconds
......@@ -285,6 +285,11 @@ class neighbor_latent_entropy(tf.keras.callbacks.Callback):
writer = tf.summary.create_file_writer(self.log_dir)
with writer.as_default():
tf.summary.scalar(
"number_of_populated_clusters",
data=len(set(hard_groups)),
step=epoch,
)
tf.summary.scalar(
"average_neighborhood_cluster_entropy",
data=np.average(purity_vector, weights=max_groups[random_idxs]),
......
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