Update phishing_email_detection_gpt2.py

david-thrower · Dec 7, 2023 · dbd6035 · dbd6035
1 parent baf9dfc
commit dbd6035
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/phishing_email_detection_gpt2.py b/phishing_email_detection_gpt2.py
@@ -166,8 +166,8 @@ def from_config(cls, config):
 epochs = 10  # [1, 100]
 batch_size = 20 # Original 20
 maximum_levels = 4  # [3,7]
-maximum_units_per_level = 8  # [2,10]
-maximum_neurons_per_unit = 5  # [2,20]
+maximum_units_per_level = 15  # [2,10]
+maximum_neurons_per_unit = 7  # [2,20]
 
 #
 # Logging