Skip to content

Commit

Permalink
j028
Browse files Browse the repository at this point in the history
  • Loading branch information
nmcardoso committed Aug 23, 2023
1 parent 1937843 commit 07bd446
Showing 1 changed file with 2 additions and 1 deletion.
3 changes: 2 additions & 1 deletion mergernet/jobs/j028.py
Original file line number Diff line number Diff line change
Expand Up @@ -32,12 +32,13 @@ def call(self):
HP.const('dense_1_units', 400),
HP.const('batch_norm_1', True),
HP.const('dropout_1_rate', 0.32),
HP.const('features_reduction', 'avg_pooling'),
# metrics and validation
HP.const('metrics', ['f1', 'recall', 'roc']),
HP.const('positive_class_id', 1),
HP.const('negative_class_id', 0),
# t1 setup
HP.const('tl_epochs', 15),
HP.const('t1_epochs', 15),
HP.const('t1_opt', 'adamw'),
HP.const('t1_lr', 5e-4),
# main train setup
Expand Down

0 comments on commit 07bd446

Please sign in to comment.