Skip to content

Commit d5bdbce

Browse files
Update phishing_email_detection_gpt2.py
Temporary test to fast forward to cerebros model.
1 parent 407f90c commit d5bdbce

File tree

1 file changed

+4
-0
lines changed

1 file changed

+4
-0
lines changed

phishing_email_detection_gpt2.py

Lines changed: 4 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -137,6 +137,8 @@ def from_config(cls, config):
137137
binary_output = tf.keras.layers.Dense(1, activation='sigmoid')(gpt2_layer)
138138
gpt_baseline_model = Model(inputs=input_layer, outputs=binary_output)
139139

140+
## Un - string out this
141+
"""
140142
gpt_baseline_model.compile(
141143
optimizer=Adam(learning_rate=1e-4), # Small LR since we're fine-tuning GPT
142144
loss='binary_crossentropy',
@@ -176,6 +178,8 @@ def from_config(cls, config):
176178
hy_df = pd.DataFrame(history.history)
177179
print(hy_df)
178180
181+
"""
182+
179183
### Cerebros model:
180184

181185
# TokenizerLayer class to handle tokenization and return only token_ids

0 commit comments

Comments
 (0)