Spaces:
Sleeping
Sleeping
Update app.py
Browse files
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 |
-
|
|
|
|
|
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()
|