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
5514e9b2
Commit
5514e9b2
authored
Apr 16, 2020
by
Andrei-Claudiu Roibu
🖥
Browse files
added data_shuffling flag and debugged run
parent
a2b51e13
Changes
2
Hide whitespace changes
Inline
Side-by-side
run.py
View file @
5514e9b2
...
...
@@ -360,35 +360,34 @@ if __name__ == '__main__':
data_parameters
[
'data_directory'
],
data_parameters
[
'train_percentage'
],
data_parameters
[
'validation_percentage'
])
update_shuffling_flag
(
'settings.ini'
)
# TODO: This might also be a very good point to add cross-validation later
else
:
if
arguments
.
mode
==
'train'
:
train
(
data_parameters
,
training_parameters
,
network_parameters
,
misc_parameters
)
# NOTE: THE EVAL FUNCTIONS HAVE NOT YET BEEN DEBUGGED (16/04/20)
elif
arguments
.
mode
==
'evaluate-score'
:
evaluate_score
(
training_parameters
,
network_parameters
,
misc_parameters
,
evaluation_parameters
)
elif
arguments
.
mode
==
'evaluate-mapping'
:
logging
.
basicConfig
(
filename
=
'evaluate-mapping-error.log'
)
if
arguments
.
settings_path
is
not
None
:
settings_evaluation
=
Settings
(
arguments
.
settings_path
)
else
:
settings_evaluation
=
Settings
(
'settings_evaluation.ini'
)
mapping_evaluation_parameters
=
settings_evaluation
[
'MAPPING'
]
evaluate_mapping
(
mapping_evaluation_parameters
)
elif
arguments
.
mode
==
'clear-experiments'
:
shutil
.
rmtree
(
os
.
path
.
join
(
misc_parameters
[
'experiments_directory'
],
training_parameters
[
'experiment_name'
]))
shutil
.
rmtree
(
os
.
path
.
join
(
misc_parameters
[
'logs_directory'
],
training_parameters
[
'experiment_name'
]))
print
(
'Cleared the current experiments and logs directory successfully!'
)
elif
arguments
.
mode
==
'clear-everything'
:
delete_files
(
misc_parameters
[
'experiments_directory'
])
delete_files
(
misc_parameters
[
'logs_directory'
])
print
(
'Cleared the current experiments and logs directory successfully!'
)
if
arguments
.
mode
==
'train'
:
train
(
data_parameters
,
training_parameters
,
network_parameters
,
misc_parameters
)
# NOTE: THE EVAL FUNCTIONS HAVE NOT YET BEEN DEBUGGED (16/04/20)
elif
arguments
.
mode
==
'evaluate-score'
:
evaluate_score
(
training_parameters
,
network_parameters
,
misc_parameters
,
evaluation_parameters
)
elif
arguments
.
mode
==
'evaluate-mapping'
:
logging
.
basicConfig
(
filename
=
'evaluate-mapping-error.log'
)
if
arguments
.
settings_path
is
not
None
:
settings_evaluation
=
Settings
(
arguments
.
settings_path
)
else
:
raise
ValueError
(
'Invalid mode value! Only supports: train, evaluate-score, evaluate-mapping, clear-experiments and clear-everything'
)
settings_evaluation
=
Settings
(
'settings_evaluation.ini'
)
mapping_evaluation_parameters
=
settings_evaluation
[
'MAPPING'
]
evaluate_mapping
(
mapping_evaluation_parameters
)
elif
arguments
.
mode
==
'clear-experiments'
:
shutil
.
rmtree
(
os
.
path
.
join
(
misc_parameters
[
'experiments_directory'
],
training_parameters
[
'experiment_name'
]))
shutil
.
rmtree
(
os
.
path
.
join
(
misc_parameters
[
'logs_directory'
],
training_parameters
[
'experiment_name'
]))
print
(
'Cleared the current experiments and logs directory successfully!'
)
elif
arguments
.
mode
==
'clear-everything'
:
delete_files
(
misc_parameters
[
'experiments_directory'
])
delete_files
(
misc_parameters
[
'logs_directory'
])
print
(
'Cleared the current experiments and logs directory successfully!'
)
else
:
raise
ValueError
(
'Invalid mode value! Only supports: train, evaluate-score, evaluate-mapping, clear-experiments and clear-everything'
)
settings.ini
View file @
5514e9b2
[DATA]
data_directory
=
"../well/win-biobank/projects/imaging/data/data3/subjectsAll/"
data_split_flag
=
Fals
e
data_split_flag
=
Tru
e
train_percentage
=
90
validation_percentage
=
5
train_list
=
"train.txt"
...
...
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