Skip to content
GitLab
Menu
Projects
Groups
Snippets
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in
Toggle navigation
Menu
On Thursday, 7th July from 1 to 3 pm there will be a maintenance with a short downtime of GitLab.
Open sidebar
Lucas Miranda
deepOF
Commits
aa6dfb9f
Commit
aa6dfb9f
authored
Mar 12, 2021
by
lucas_miranda
Browse files
Implemented KNN_purity callback
parent
5f256847
Pipeline
#95550
passed with stages
in 60 minutes and 35 seconds
Changes
1
Pipelines
1
Hide whitespace changes
Inline
Side-by-side
deepof/train_utils.py
View file @
aa6dfb9f
...
...
@@ -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
]
...
...
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment