Commit 330bbd80 authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented KNN_purity callback

parent 3c773f9f
Pipeline #95527 failed with stages
in 23 minutes and 46 seconds
......@@ -239,8 +239,6 @@ class knn_cluster_purity(tf.keras.callbacks.Callback):
self.model.layers[0].input, cluster_assignment.output
)
print(self.validation_data)
# Use encoder and grouper to predict on validation data
encoding = encoder.predict(self.validation_data)
groups = grouper.predict(self.validation_data)
......
......@@ -116,6 +116,7 @@ def get_callbacks(
k=knn_neighbors,
samples=knn_samples,
validation_data=X_val,
log_dir=os.path.join(outpath, "metrics"),
)
onecycle = deepof.model_utils.one_cycle_scheduler(
......
Supports Markdown
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