j027

nmcardoso · Aug 19, 2023 · 6cc3495 · 6cc3495
1 parent 5495ac0
commit 6cc3495
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/mergernet/jobs/j027.py b/mergernet/jobs/j027.py
@@ -27,7 +27,7 @@ def call(self):
     hps = HyperParameterSet(
       HP.const('architecture', 'efficientnetv2b0'),
       HP.const('pretrained_weights', 'imagenet'),
-      HP.const('metrics', ['f1', 'recall']),
+      HP.const('metrics', ['f1', 'recall', 'roc']),
       HP.const('positive_class_id', 1),
       HP.const('negative_class_id', 0),
       HP.const('epochs', 35),