From 0bddc2da7e0c693009f49eaf34268dff8fd99aec Mon Sep 17 00:00:00 2001 From: David Thrower Date: Wed, 6 Dec 2023 16:51:33 -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 f7809ea..81a31e8 100644 --- 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]