Update phishing_email_detection_gpt2.py

david-thrower · web-flow · commit 407f90cb8759 · 2025-03-22T18:54:18.000-04:00
Correct history indexing error.
diff --git a/phishing_email_detection_gpt2.py b/phishing_email_detection_gpt2.py
@@ -173,8 +173,7 @@ def from_config(cls, config):
 gpt_t1 = time.time()
 gpt_time_on_one_model_min =  (gpt_t1 - gpt_t1) / 60
 
-hy = history["history"]
-hy_df = pd.DataFrame(hy)
+hy_df = pd.DataFrame(history.history)
 print(hy_df)
 
 ### Cerebros model:
