From ec0ee7d3766876e8c659ed2e37a63d03aa23bec5 Mon Sep 17 00:00:00 2001 From: David Thrower Date: Wed, 21 Feb 2024 20:07:45 -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 c231415..c0f9646 100644 --- a/phishing_email_detection_gpt2.py +++ b/phishing_email_detection_gpt2.py @@ -123,7 +123,7 @@ # GPT2 configurables -max_seq_length = 900 +max_seq_length = 2000 # inp = tf.keras.layers.Input(shape=(), dtype=tf.string) # gp2 = GPT2Layer(max_seq_length=max_seq_length) @@ -179,7 +179,7 @@ def call(self, text): output_dim=18, input_length=max_seq_length)(tokens_1) flat = tf.keras.layers.Flatten()(embedded) -dropout_flat = tf.keras.layers.Dropout(0.6)(flat) +dropout_flat = tf.keras.layers.Dropout(0.75)(flat) tokenized_embedded_model =\ tf.keras.Model(