Skip to content

Commit a480dfd

Browse files
Update phishing_email_detection_gpt2.py
Correct metric to rank by (binary accuracy) ...
1 parent 0b67f88 commit a480dfd

File tree

1 file changed

+1
-1
lines changed

1 file changed

+1
-1
lines changed

phishing_email_detection_gpt2.py

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -279,7 +279,7 @@ def from_config(cls, config):
279279
labels=train_labels,
280280
validation_split=0.35,
281281
direction='maximize',
282-
metric_to_rank_by="val_accuracy",
282+
metric_to_rank_by="val_binary_accuracy",
283283
minimum_levels=minimum_levels,
284284
maximum_levels=maximum_levels,
285285
minimum_units_per_level=minimum_units_per_level,

0 commit comments

Comments
 (0)