Skip to content

Commit 2759e1c

Browse files
Update phishing_email_detection_gpt2.py
Set NAS to minimize Perplexity.
1 parent ac6c295 commit 2759e1c

File tree

1 file changed

+2
-2
lines changed

1 file changed

+2
-2
lines changed

phishing_email_detection_gpt2.py

Lines changed: 2 additions & 2 deletions
Original file line numberDiff line numberDiff line change
@@ -570,8 +570,8 @@ def reset_state(self):
570570
training_data=x_train_packaged,
571571
labels=y_train_packaged,
572572
validation_split=0.2,
573-
direction='maximize',
574-
metric_to_rank_by="val_categorical_accuracy",
573+
direction='minimize',
574+
metric_to_rank_by="perplexity", # REPLACE WITH VAL PERPLEXITY ONCE THIS IS STABLE
575575
minimum_levels=minimum_levels,
576576
maximum_levels=maximum_levels,
577577
minimum_units_per_level=minimum_units_per_level,

0 commit comments

Comments
 (0)