modif retrain metrics neon

retrain.py

@@ -14,6 +14,7 @@ import joblib
 import pickle
 import xgboost as xgb
 import smtplib
+import datetime
 from email.mime.text import MIMEText
 from email.mime.multipart import MIMEMultipart
 from email.utils import formataddr
@@ -252,6 +253,13 @@ def run_retraining():
 
         print("Email envoyé avec succès.")
 
+    def load_metrics_neon(c_index):
+        raw = {"date":datetime.today(), "resultat":c_index, "status":"retrain_model"}
+        df_raw = pd.Dataframe([raw]) 
+        df_raw["date"] = pd.to.datetime(df_raw["date"])
+        df_raw.to_sql("metrics", engine, index=False)
+
+
 
     if __name__ == "__main__":
         xgb_final = pipeline
@@ -260,4 +268,7 @@ def run_retraining():
         )
         print(f"Run ID: {run_id}")
         send_email_cindex(c_index, model_path)
+        print("Email envoyé")
+        load_metrics_neon(c_index)
+        print("Metrics envoyée sur neon")