Update phishing_email_detection_gpt2.py

david-thrower · Dec 6, 2023 · 0bddc2d · 0bddc2d
1 parent 2145a5f
commit 0bddc2d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/phishing_email_detection_gpt2.py b/phishing_email_detection_gpt2.py
@@ -162,9 +162,9 @@ def from_config(cls, config):
 max_consecutive_lateral_connections = 22
 p_lateral_connection = 0.39256
 num_lateral_connection_tries_per_unit = 10
-learning_rate = 0.000025 # original 0.0000511065
+learning_rate = 0.0000511065 # original 0.0000511065
 epochs = 15  # [1, 100]
-batch_size = 30 # Original 20
+batch_size = 15 # Original 20
 maximum_levels = 4  # [3,7]
 maximum_units_per_level = 8  # [2,10]
 maximum_neurons_per_unit = 5  # [2,20]