vatsal-p-wa commited on
Commit
03281b1
·
verified ·
1 Parent(s): 291692c

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -228,7 +228,9 @@ def process_dataframe(df):
228
  # Transform categorical columns for prediction DataFrame using the label encoders.
229
  for col in ['Tag', 'EngShp', 'EngQua', 'EngCol', 'EngCut', 'EngPol', 'EngSym', 'EngFlo', 'EngNts', 'EngMikly','EngBlk', 'EngWht', 'EngOpen', 'EngPav']:
230
  try:
231
- df_pred[col] = loaded_label_encoder[col].transform(df_pred[col])
 
 
232
  except ValueError as e:
233
  print(f'Invalid value in column {col}: {e}', 'error')
234
  return pd.DataFrame(), pd.DataFrame()
 
228
  # Transform categorical columns for prediction DataFrame using the label encoders.
229
  for col in ['Tag', 'EngShp', 'EngQua', 'EngCol', 'EngCut', 'EngPol', 'EngSym', 'EngFlo', 'EngNts', 'EngMikly','EngBlk', 'EngWht', 'EngOpen', 'EngPav']:
230
  try:
231
+ encoder = loaded_label_encoder[col]
232
+ df_pred[col] = df_pred[col].map(lambda x: encoder.transform([x])[0] if x in encoder.classes_ else -1)
233
+ # df_pred[col] = loaded_label_encoder[col].transform(df_pred[col])
234
  except ValueError as e:
235
  print(f'Invalid value in column {col}: {e}', 'error')
236
  return pd.DataFrame(), pd.DataFrame()