From baf9dfc56a38a9ec5f08ad30710fa9bb62bc1b60 Mon Sep 17 00:00:00 2001 From: David Thrower Date: Thu, 7 Dec 2023 13:01:15 -0500 Subject: [PATCH] Update phishing_email_detection_gpt2.py --- phishing_email_detection_gpt2.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/phishing_email_detection_gpt2.py b/phishing_email_detection_gpt2.py index b3143c4..efda09c 100644 --- a/phishing_email_detection_gpt2.py +++ b/phishing_email_detection_gpt2.py @@ -163,7 +163,7 @@ def from_config(cls, config): p_lateral_connection = 0.39256 num_lateral_connection_tries_per_unit = 10 learning_rate = 0.0000511065 # original 0.0000511065 -epochs = 15 # [1, 100] +epochs = 10 # [1, 100] batch_size = 20 # Original 20 maximum_levels = 4 # [3,7] maximum_units_per_level = 8 # [2,10]