Update src/module/Features.py

src/module/Features.py

@@ -348,8 +348,7 @@ class FeatureEngineer:
         master = self.create_interaction_features(master)
         
         # ===== 8. NETTOYAGE FINAL LÉGER =====
-        # On supprime SEULEMENT les lignes où les lags sont NaN
-        # (premières lignes de chaque groupe)
+        # On supprime SEULEMENT les lignes où les lags sont NaN (premières lignes de chaque groupe)
         rows_before = len(master)
         master = master.dropna()
         rows_after = len(master)
@@ -358,12 +357,32 @@ class FeatureEngineer:
         if rows_removed > 0:
             st.info(f"ℹ️ {rows_removed} lignes supprimées (NaN des lags sur premières dates)")
         
+        # =========================================================
+        # 🔥 AJOUT CRITIQUE : ENCODAGE POUR COMPATIBILITÉ ML 🔥
+        # =========================================================
+        # Les modèles (XGBoost, Random Forest) ne savent pas lire le texte (ex: "CHIRURGIE").
+        # On convertit toutes les colonnes TEXTE (sauf la date) en CHIFFRES.
+        
+        # On détecte les colonnes de type 'object' (texte) ou 'category'
+        categorical_cols = master.select_dtypes(include=['object', 'category']).columns
+        
+        count_encoded = 0
+        for col in categorical_cols:
+            if col != 'date':  # On ne touche SURTOUT PAS à la date (utile pour le split temporel)
+                # Transformation : "CHIRURGIE" -> 0, "REANIMATION" -> 1
+                master[col] = master[col].astype('category').cat.codes
+                count_encoded += 1
+
+        if count_encoded > 0:
+            st.caption(f"🔧 {count_encoded} colonnes texte converties en codes numériques pour l'IA.")
+
+        # =========================================================
+
         # Stockage du dataset final
         self.ml_dataset = master
         
         return master, f"✅ Dataset ML créé : {len(master):,} échantillons, {len(master.columns)} features"
 
-
 # =====================================================
 # ============= INTERFACE STREAMLIT ===================
 # =====================================================