Skip to content
Projects
Groups
Snippets
Help
Loading...
Help
Support
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
D
DeepPPI
Project
Project
Details
Activity
Releases
Cycle Analytics
Repository
Repository
Files
Commits
Branches
Tags
Contributors
Graph
Compare
Charts
Issues
0
Issues
0
List
Boards
Labels
Milestones
Merge Requests
0
Merge Requests
0
CI / CD
CI / CD
Pipelines
Jobs
Schedules
Charts
Packages
Packages
Container Registry
Wiki
Wiki
Snippets
Snippets
Members
Members
Collapse sidebar
Close sidebar
Activity
Graph
Charts
Create a new issue
Jobs
Commits
Issue Boards
Open sidebar
Florian RICHOUX
DeepPPI
Commits
186060a3
Commit
186060a3
authored
Jan 04, 2019
by
Florian RICHOUX
1
Browse files
Options
Browse Files
Download
Email Patches
Plain Diff
Verbose for callbacks
parent
7a6eb308
Changes
1
Hide whitespace changes
Inline
Side-by-side
Showing
1 changed file
with
8 additions
and
6 deletions
+8
-6
main.py
keras/main.py
+8
-6
No files found.
keras/main.py
View file @
186060a3
...
...
@@ -188,18 +188,20 @@ if __name__ == '__main__':
val_p1
,
val_p2
,
val_labels
=
dl
.
load_data
(
validation_set
)
if
test_set
:
callbacks_list
=
[
callbacks
.
ReduceLROnPlateau
(
monitor
=
'loss'
,
factor
=
0.3
,
patience
=
5
)
]
callbacks_list
.
append
(
callbacks
.
EarlyStopping
(
monitor
=
'acc'
,
patience
=
patience
,)
)
#callbacks_list = []
callbacks_list
=
[
callbacks
.
ReduceLROnPlateau
(
monitor
=
'loss'
,
factor
=
0.5
,
patience
=
5
,
min_lr
=
0.0003
,
cooldown
=
1
,
verbose
=
1
)
]
callbacks_list
.
append
(
callbacks
.
EarlyStopping
(
monitor
=
'acc'
,
patience
=
patience
,
verbose
=
1
)
)
if
save_weights
:
callbacks_list
.
append
(
callbacks
.
ModelCheckpoint
(
filepath
=
'weights/'
+
file_name
+
'.h5'
,
monitor
=
'loss'
,
save_best_only
=
True
,)
)
monitor
=
'loss'
,
save_best_only
=
True
,
verbose
=
1
)
)
else
:
callbacks_list
=
[
callbacks
.
ReduceLROnPlateau
(
monitor
=
'val_loss'
,
factor
=
0.3
,
patience
=
5
)
]
callbacks_list
.
append
(
callbacks
.
EarlyStopping
(
monitor
=
'val_acc'
,
patience
=
patience
,)
)
#callbacks_list = []
callbacks_list
=
[
callbacks
.
ReduceLROnPlateau
(
monitor
=
'val_loss'
,
factor
=
0.5
,
patience
=
5
,
min_lr
=
0.0003
,
cooldown
=
1
,
verbose
=
1
)
]
callbacks_list
.
append
(
callbacks
.
EarlyStopping
(
monitor
=
'val_acc'
,
patience
=
patience
,
verbose
=
1
)
)
#callbacks_list = [ callbacks.EarlyStopping(monitor='val_loss', patience=patience,) ]
if
save_weights
:
callbacks_list
.
append
(
callbacks
.
ModelCheckpoint
(
filepath
=
'weights/'
+
file_name
+
'.h5'
,
monitor
=
'val_loss'
,
save_best_only
=
True
,)
)
monitor
=
'val_loss'
,
save_best_only
=
True
,
verbose
=
1
)
)
else
:
val_p1
,
val_p2
,
val_labels
=
None
,
None
,
None
...
...
Florian RICHOUX
@richoux-f
mentioned in commit
c0371b3c
·
Jan 06, 2019
mentioned in commit
c0371b3c
mentioned in commit c0371b3c642e9d7c7e376c736d90aa709a237619
Toggle commit list
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