Update phishing_email_detection_gpt2.py

david-thrower · Dec 8, 2023 · 33531e8 · 33531e8
1 parent 6cd434c
commit 33531e8
Showing 1 changed file with 1 addition and 1 deletion.
diff --git 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 = 10  # [1, 100]
+epochs = 15  # [1, 100]
 batch_size = 20 # Original 20
 maximum_levels = 4  # [3,7]
 maximum_units_per_level = 8  # [2,10]