Skip to content

Commit 048eb1b

Browse files
Update phishing_email_detection_gpt2.py
Corrected hard set on levels to correct optima.
1 parent cdb4455 commit 048eb1b

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
@@ -274,8 +274,8 @@ def from_config(cls, config):
274274
learning_rate = 0.0000511065
275275
epochs = 15 # [1, 100]
276276
batch_size = 20
277-
minimum_levels = 3
278-
maximum_levels = 3 # [3,7]
277+
minimum_levels = 4
278+
maximum_levels = 4 # [3,7]
279279

280280
minimum_units_per_level = 4
281281
maximum_units_per_level = 8

0 commit comments

Comments
 (0)