Update phishing_email_detection_gpt2.py

david-thrower · Dec 8, 2023 · 233e882 · 233e882
1 parent 14b8c9b
commit 233e882
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/phishing_email_detection_gpt2.py b/phishing_email_detection_gpt2.py
@@ -167,7 +167,7 @@ def from_config(cls, config):
 minimum_levels = 2
 maximum_levels = 4 # [3,7]
 
-minimum_units_per_level = 5 # 4
+minimum_units_per_level = 4
 maximum_units_per_level = 8
 
 minimum_neurons_per_unit = 1