Spaces:
Sleeping
Sleeping
albin
commited on
Commit
·
f5e9881
1
Parent(s):
0f41c4f
correct bugs
Browse files
main.py
CHANGED
|
@@ -85,9 +85,8 @@ async def predict(request: Request, requess: Req = Depends(form_req)):
|
|
| 85 |
|
| 86 |
# Apply features encoding (convert everything into int64)
|
| 87 |
for column in selected_columns:
|
| 88 |
-
|
| 89 |
-
|
| 90 |
-
dataFrame_features[column] = encoder.transform(dataFrame_features[column])
|
| 91 |
|
| 92 |
data = dataFrame_features[selected_columns].values
|
| 93 |
|
|
|
|
| 85 |
|
| 86 |
# Apply features encoding (convert everything into int64)
|
| 87 |
for column in selected_columns:
|
| 88 |
+
encoder = label_encoders[column]
|
| 89 |
+
dataFrame_features[column] = encoder.transform(dataFrame_features[column])
|
|
|
|
| 90 |
|
| 91 |
data = dataFrame_features[selected_columns].values
|
| 92 |
|