fendy07 commited on
Commit
5fc8639
·
1 Parent(s): 220c523

update variables to reflect new model path

Browse files
Files changed (1) hide show
  1. 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['shoppsing_mall'])
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