Spaces:
Sleeping
Sleeping
update variables to reflect new model path
Browse files- src/pages/predict.py +1 -1
src/pages/predict.py
CHANGED
|
@@ -55,7 +55,7 @@ y = retail[['category']]
|
|
| 55 |
le = LabelEncoder()
|
| 56 |
X['gender'] = le.fit_transform(X['gender'])
|
| 57 |
X['payment_method'] = le.fit_transform(X['payment_method'])
|
| 58 |
-
X['shopping_mall'] = le.fit_transform(X['
|
| 59 |
y_encoded = le.fit_transform(y)
|
| 60 |
|
| 61 |
# Splitting data
|
|
|
|
| 55 |
le = LabelEncoder()
|
| 56 |
X['gender'] = le.fit_transform(X['gender'])
|
| 57 |
X['payment_method'] = le.fit_transform(X['payment_method'])
|
| 58 |
+
X['shopping_mall'] = le.fit_transform(X['shopping_mall'])
|
| 59 |
y_encoded = le.fit_transform(y)
|
| 60 |
|
| 61 |
# Splitting data
|