Update phishing_email_detection_gpt2.py

david-thrower · web-flow · commit f229a476ac63 · 2025-04-09T18:40:52.000-04:00
...
diff --git a/phishing_email_detection_gpt2.py b/phishing_email_detection_gpt2.py
@@ -74,11 +74,11 @@ def rotate_half(x):
 #
 
 # Training data for baseline model
-# baseline_train_x = tf.constant(X_train)
+baseline_train_x = tf.constant(X_train)
 baseline_train_y = tf.constant(y_train, dtype=tf.int8)
 
 # Packaged for Cerebros (multimodal, takes inputs as a list)
-training_x   = [X_train.tolist()]
+training_x   = [baseline_train_x]
 train_labels = [baseline_train_y]
 
 #

Original file line number	Diff line number	Diff line change
`@@ -74,11 +74,11 @@ def rotate_half(x):`
`74`	`74`	`#`
`75`	`75`
`76`	`76`	`# Training data for baseline model`
`77`		`-# baseline_train_x = tf.constant(X_train)`
	`77`	`+baseline_train_x = tf.constant(X_train)`
`78`	`78`	`baseline_train_y = tf.constant(y_train, dtype=tf.int8)`
`79`	`79`
`80`	`80`	`# Packaged for Cerebros (multimodal, takes inputs as a list)`
`81`		`-training_x = [X_train.tolist()]`
	`81`	`+training_x = [baseline_train_x]`
`82`	`82`	`train_labels = [baseline_train_y]`
`83`	`83`
`84`	`84`	`#`