Spaces:
Sleeping
Sleeping
Upload processing.py
Browse files- 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 |
-
|
| 70 |
-
|
| 71 |
-
|
| 72 |
-
|
| 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)
|