Skip to content
Snippets Groups Projects
Commit aa6dfb9f authored by Lucas Miranda's avatar Lucas Miranda
Browse files

Implemented KNN_purity callback

parent 5f256847
Branches
Tags
No related merge requests found
Pipeline #95550 passed
...@@ -116,14 +116,14 @@ def get_callbacks( ...@@ -116,14 +116,14 @@ def get_callbacks(
k=knn_neighbors, k=knn_neighbors,
samples=knn_samples, samples=knn_samples,
validation_data=X_val, validation_data=X_val,
log_dir=os.path.join(outpath, "metrics"), log_dir=os.path.join(outpath, "metrics", run_ID),
variational=variational, variational=variational,
) )
onecycle = deepof.model_utils.one_cycle_scheduler( onecycle = deepof.model_utils.one_cycle_scheduler(
X_train.shape[0] // batch_size * 250, X_train.shape[0] // batch_size * 250,
max_rate=0.005, 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] callbacks = [run_ID, tensorboard_callback, knn, onecycle]
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please register or to comment