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
Open sidebar
Andrei-Claudiu Roibu
BrainMapper
Commits
da80e6cb
Commit
da80e6cb
authored
Apr 20, 2020
by
Andrei-Claudiu Roibu
🖥
Browse files
changed test logs name to validation
parent
4c63df04
Changes
1
Hide whitespace changes
Inline
Side-by-side
utils/data_logging_utils.py
View file @
da80e6cb
...
@@ -53,19 +53,19 @@ class LogWriter():
...
@@ -53,19 +53,19 @@ class LogWriter():
self
.
number_of_classes
=
number_of_classes
self
.
number_of_classes
=
number_of_classes
training_logs_directory
=
os
.
path
.
join
(
training_logs_directory
=
os
.
path
.
join
(
logs_directory
,
experiment_name
,
"train"
)
logs_directory
,
experiment_name
,
"train"
)
testing
_logs_directory
=
os
.
path
.
join
(
validation
_logs_directory
=
os
.
path
.
join
(
logs_directory
,
experiment_name
,
"
test
"
)
logs_directory
,
experiment_name
,
"
validation
"
)
# If the logs directory exist, we clear their contents to allow new logs to be created
# If the logs directory exist, we clear their contents to allow new logs to be created
if
not
use_last_checkpoint
:
if
not
use_last_checkpoint
:
if
os
.
path
.
exists
(
training_logs_directory
):
if
os
.
path
.
exists
(
training_logs_directory
):
shutil
.
rmtree
(
training_logs_directory
)
shutil
.
rmtree
(
training_logs_directory
)
if
os
.
path
.
exists
(
testing
_logs_directory
):
if
os
.
path
.
exists
(
validation
_logs_directory
):
shutil
.
rmtree
(
testing
_logs_directory
)
shutil
.
rmtree
(
validation
_logs_directory
)
self
.
log_writer
=
{
self
.
log_writer
=
{
'train'
:
SummaryWriter
(
log_dir
=
training_logs_directory
),
'train'
:
SummaryWriter
(
log_dir
=
training_logs_directory
),
'
test
'
:
SummaryWriter
(
log_dir
=
testing
_logs_directory
)
'
validation
'
:
SummaryWriter
(
log_dir
=
validation
_logs_directory
)
}
}
self
.
confusion_matrix_color_map
=
confusion_matrix_cmap
self
.
confusion_matrix_color_map
=
confusion_matrix_cmap
...
@@ -247,4 +247,4 @@ class LogWriter():
...
@@ -247,4 +247,4 @@ class LogWriter():
"""
"""
self
.
log_writer
[
'train'
].
close
()
self
.
log_writer
[
'train'
].
close
()
self
.
log_writer
[
'
test
'
].
close
()
self
.
log_writer
[
'
validation
'
].
close
()
Write
Preview
Supports
Markdown
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