From 8602ac60fd582375e3574997d7652e8e69a8fb10 Mon Sep 17 00:00:00 2001 From: David Thrower Date: Fri, 8 Dec 2023 00:26:53 -0500 Subject: [PATCH] Update phishing_email_detection_gpt2.py --- phishing_email_detection_gpt2.py | 4 ++-- 1 file changed, 2 insertions(+), 2 deletions(-) diff --git a/phishing_email_detection_gpt2.py b/phishing_email_detection_gpt2.py index 264f48b..c1a57c3 100644 --- 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] #