Djohell commited on
Commit
33c4a7c
·
verified ·
1 Parent(s): 96e1f6f

Upload processing.py

Browse files
Files changed (1) hide show
  1. processing.py +5 -6
processing.py CHANGED
@@ -65,18 +65,17 @@ def map_statut_expert(p2):
65
  def prepare_input(data):
66
  # On crée le DataFrame avec les colonnes exactes du modèle
67
  df = pd.DataFrame(0.0, index=[0], columns=FEATURES)
68
-
69
- # 1. Variables numériques directes
70
- # Attention au nom de la colonne âge : doit être identique à ton secret FEATURES
71
- col_age = 'age_au_diagnostic' if 'age_au_diagnostic' in FEATURES else 'age_estime'
72
- df[col_age] = float(data.get('age_estime', 0))
73
 
74
  df['Tranche_effectif_num'] = float(data.get('Tranche_effectif_num', 0))
75
  df['is_ess'] = int(data.get('is_ess', 0))
76
 
77
  # 2. GESTION DU DÉPARTEMENT (Sauvetage de la Corse)
78
  code_dep = str(data.get('code_departement', '')).strip().upper()
79
- # On cherche la valeur dans le mapping S3, sinon 0.05 par défaut
80
  df['risque_departemental'] = float(DEP_RISK_MAP.get(code_dep, 0.05))
81
 
82
  # 3. Mapping APE (Via S3)
 
65
  def prepare_input(data):
66
  # On crée le DataFrame avec les colonnes exactes du modèle
67
  df = pd.DataFrame(0.0, index=[0], columns=FEATURES)
68
+
69
+ if 'age_au_diagnostic' in FEATURES:
70
+ df['age_au_diagnostic'] = float(data.get('age_estime', 0))
71
+ else:
72
+ df['age_estime'] = float(data.get('age_estime', 0))
73
 
74
  df['Tranche_effectif_num'] = float(data.get('Tranche_effectif_num', 0))
75
  df['is_ess'] = int(data.get('is_ess', 0))
76
 
77
  # 2. GESTION DU DÉPARTEMENT (Sauvetage de la Corse)
78
  code_dep = str(data.get('code_departement', '')).strip().upper()
 
79
  df['risque_departemental'] = float(DEP_RISK_MAP.get(code_dep, 0.05))
80
 
81
  # 3. Mapping APE (Via S3)