Update phishing_email_detection_gpt2.py

david-thrower · web-flow · commit 5ce1eb86e4d9 · 2025-04-09T18:47:20.000-04:00
...
diff --git a/phishing_email_detection_gpt2.py b/phishing_email_detection_gpt2.py
@@ -74,7 +74,7 @@ def rotate_half(x):
 #
 
 # Training data for baseline model
-baseline_train_x = tf.constant(X_train)
+baseline_train_x = tf.constant(X_train, dtype=tf.string)
 baseline_train_y = tf.constant(y_train, dtype=tf.int8)
 
 # Packaged for Cerebros (multimodal, takes inputs as a list)

Original file line number	Diff line number	Diff line change
`@@ -74,7 +74,7 @@ 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, dtype=tf.string)`
`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)`