Shalyn commited on
Commit
5ed0329
·
verified ·
1 Parent(s): 0037aec

Upload folder using huggingface_hub

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -56,7 +56,7 @@ def sales_batch_predict():
56
 
57
  #convert the categorical to dummies
58
  categorical_columns_for_dummies = ['Product_Sugar_Content','Product_Type','Store_Size','Store_Location_City_Type','Store_Type']
59
- input_df_dummies = pd.get_dummies(input_data, columns=categorical_columns_for_dummies, drop_first=False)
60
  input_df_aligned =input_df_dummies.reindex(columns=X_train.columns, fill_value=0)
61
 
62
  #predict
 
56
 
57
  #convert the categorical to dummies
58
  categorical_columns_for_dummies = ['Product_Sugar_Content','Product_Type','Store_Size','Store_Location_City_Type','Store_Type']
59
+ input_df_dummies = pd.get_dummies(input_data, columns=categorical_columns_for_dummies, drop_first=True)
60
  input_df_aligned =input_df_dummies.reindex(columns=X_train.columns, fill_value=0)
61
 
62
  #predict