Skip to content

Commit 966f714

Browse files
Update phishing_email_detection_gpt2.py
Combined best hyperparams from the hyperparameter optimization study with AdamW optimizer.
1 parent 8939f3c commit 966f714

File tree

1 file changed

+13
-13
lines changed

1 file changed

+13
-13
lines changed

phishing_email_detection_gpt2.py

Lines changed: 13 additions & 13 deletions
Original file line numberDiff line numberDiff line change
@@ -223,7 +223,7 @@ def from_config(cls, config):
223223

224224
# On larger hardware, this could probably be increased considerably and
225225
# Probably would improve performance ...
226-
EMBEDDING_DIM = 15 # Define EMBEDDING_DIM here, to match your embedding layer.
226+
EMBEDDING_DIM = 23 # Define EMBEDDING_DIM here, to match your embedding layer.
227227

228228
embedded = tf.keras.layers.Embedding(
229229
input_dim=VOCABULARY_SIZE,
@@ -241,7 +241,7 @@ def from_config(cls, config):
241241
# Just an FYI for anyone trying to apply conventional wisdom
242242
# to save you the time ...
243243
x = x = tf.keras.layers.Concatenate()([embedded, position_embedding])
244-
x = tf.keras.layers.Dropout(0.6)(x) # AI suggested 0.4
244+
x = tf.keras.layers.Dropout(0.4)(x) # AI suggested 0.4
245245
flattened = tf.keras.layers.Flatten()(x)
246246

247247
cerebros_base_model = tf.keras.Model(
@@ -255,23 +255,23 @@ def from_config(cls, config):
255255
#
256256
# Cerebros configurables
257257
#
258-
activation = 'gelu'
259-
predecessor_level_connection_affinity_factor_first = 49.9999
260-
predecessor_level_connection_affinity_factor_main = 0.31456
261-
max_consecutive_lateral_connections = 22
262-
p_lateral_connection = 0.39256
263-
num_lateral_connection_tries_per_unit = 10
264-
learning_rate = 0.0000511065
258+
activation = "relu"
259+
predecessor_level_connection_affinity_factor_first = 10
260+
predecessor_level_connection_affinity_factor_main = 40
261+
max_consecutive_lateral_connections = 20
262+
p_lateral_connection = 30
263+
num_lateral_connection_tries_per_unit = 25
264+
learning_rate = 3 * 10 ** -3
265265
epochs = 15 # [1, 100]
266-
batch_size = 20
266+
batch_size = 17
267267
minimum_levels = 2
268-
maximum_levels = 3 # [3,7]
268+
maximum_levels = 2 # [3,7]
269269

270270
minimum_units_per_level = 4
271-
maximum_units_per_level = 8
271+
maximum_units_per_level = 7
272272

273273
minimum_neurons_per_unit = 1
274-
maximum_neurons_per_unit = 5 # [2,20]
274+
maximum_neurons_per_unit = 2
275275

276276
moities_to_try = 5
277277
tries_per_moity = 1

0 commit comments

Comments
 (0)