Update phishing_email_detection_gpt2.py

david-thrower · Dec 8, 2023 · 8602ac6 · 8602ac6
1 parent 33531e8
commit 8602ac6
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/phishing_email_detection_gpt2.py b/phishing_email_detection_gpt2.py
@@ -165,8 +165,8 @@ def from_config(cls, config):
 learning_rate = 0.0000511065 # original 0.0000511065
 epochs = 15  # [1, 100]
 batch_size = 20 # Original 20
-maximum_levels = 4  # [3,7]
-maximum_units_per_level = 8  # [2,10]
+maximum_levels = 3 # 4 was best  # [3,7]
+maximum_units_per_level = 7 # 8 was best
 maximum_neurons_per_unit = 5  # [2,20]
 
 #