Commit aa6dfb9f authored by lucas_miranda's avatar lucas_miranda
Browse files

Implemented KNN_purity callback

parent 5f256847
Pipeline #95550 passed with stages
in 60 minutes and 35 seconds
......@@ -116,14 +116,14 @@ def get_callbacks(
k=knn_neighbors,
samples=knn_samples,
validation_data=X_val,
log_dir=os.path.join(outpath, "metrics"),
log_dir=os.path.join(outpath, "metrics", run_ID),
variational=variational,
)
onecycle = deepof.model_utils.one_cycle_scheduler(
X_train.shape[0] // batch_size * 250,
max_rate=0.005,
log_dir=os.path.join(outpath, "metrics"),
log_dir=os.path.join(outpath, "metrics", run_ID),
)
callbacks = [run_ID, tensorboard_callback, knn, onecycle]
......
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