Update phishing_email_detection_gpt2.py

david-thrower · Feb 22, 2024 · ec0ee7d · ec0ee7d
1 parent 14c99a6
commit ec0ee7d
Showing 1 changed file with 2 additions and 2 deletions.
diff --git 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(