gdleds commited on
Commit
83eee1d
Β·
verified Β·
1 Parent(s): a7bb76c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -219,8 +219,8 @@ def predict(df_raw: pd.DataFrame, model, baseline_survival) -> pd.DataFrame:
219
  for t, col in horizons.items():
220
  S_t = (S0_at(t)) ** HR # survival
221
  df[col] = 1 - S_t # prob event
222
-
223
- df_map = df[["latitude_feu", "longitude_feu", "ville"] + list(horizons.values())].copy()
224
  return df_map
225
 
226
  # ────────────────────────────────────────────────
 
219
  for t, col in horizons.items():
220
  S_t = (S0_at(t)) ** HR # survival
221
  df[col] = 1 - S_t # prob event
222
+ df_map = df[["lat", "lon", "nom"] + list(horizons.values())].copy()
223
+ # df_map = df[["latitude_feu", "longitude_feu", "ville"] + list(horizons.values())].copy()
224
  return df_map
225
 
226
  # ────────────────────────────────────────────────