zotthytt12 commited on
Commit
01828f7
·
verified ·
1 Parent(s): b6b5513

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -140,8 +140,8 @@ def rank_candidates(request: RankingRequest):
140
  # Upewnij się, że brakuje tylko kolumny 'identifier', a reszta pasuje
141
  features_df = df.drop(columns=['identifier'])
142
 
143
- # Ustaw kolejność kolumn DOKŁADNIE tak, jak w treningu
144
- features_df_ordered = features_df[MODEL_FEATURES_ORDER]
145
 
146
  # 3. Predykcja
147
  probabilities = model.predict_proba(features_df_ordered)[:, 1]
 
140
  # Upewnij się, że brakuje tylko kolumny 'identifier', a reszta pasuje
141
  features_df = df.drop(columns=['identifier'])
142
 
143
+ features_df_ordered = features_df.reindex(columns=model.feature_names_in_, fill_value=0)
144
+
145
 
146
  # 3. Predykcja
147
  probabilities = model.predict_proba(features_df_ordered)[:, 1]