Skip to content

Commit 49a16c6

Browse files
Update phishing_email_detection_gpt2.py
Try levels hard set to 2.
1 parent 7930a2d commit 49a16c6

File tree

1 file changed

+3
-13
lines changed

1 file changed

+3
-13
lines changed

phishing_email_detection_gpt2.py

Lines changed: 3 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -85,7 +85,7 @@
8585

8686
"""### A custom GPT2 encoder layer for text embedding"""
8787

88-
88+
"""
8989
class GPT2Layer(tf.keras.layers.Layer):
9090
9191
def __init__(self, max_seq_length, **kwargs):
@@ -182,7 +182,7 @@ def from_config(cls, config):
182182
183183
hy_df = pd.DataFrame(history.history)
184184
print(hy_df)
185-
185+
"""
186186

187187
### Cerebros model:
188188

@@ -249,16 +249,6 @@ def from_config(cls, config):
249249

250250

251251

252-
253-
254-
# dropout_embedded = tf.keras.layers.Dropout(0.6)(embedded)
255-
# flattened = tf.keras.layers.Flatten()(dropout_embedded)
256-
257-
# cerebros_base_model =\
258-
# tf.keras.Model(
259-
# inputs=inp,
260-
# outputs=flattened)
261-
262252
"""### Cerebros search for the best model"""
263253

264254
#
@@ -274,7 +264,7 @@ def from_config(cls, config):
274264
epochs = 15 # [1, 100]
275265
batch_size = 20
276266
minimum_levels = 2
277-
maximum_levels = 3 # [3,7]
267+
maximum_levels = 2 # [3,7]
278268

279269
minimum_units_per_level = 4
280270
maximum_units_per_level = 8

0 commit comments

Comments
 (0)