Upload folder using huggingface_hub
Browse files
app.py
CHANGED
|
@@ -72,7 +72,7 @@ def sales_predict():
|
|
| 72 |
input_df_dummies = pd.get_dummies(input_data, columns=categorical_columns_for_dummies, drop_first=True)
|
| 73 |
|
| 74 |
# Reindex input_df_dummies to match the columns of X_train used during training
|
| 75 |
-
input_df_aligned = input_df_dummies.reindex(columns=
|
| 76 |
|
| 77 |
#make model to predict
|
| 78 |
prediction = model.predict(input_df_aligned)
|
|
|
|
| 72 |
input_df_dummies = pd.get_dummies(input_data, columns=categorical_columns_for_dummies, drop_first=True)
|
| 73 |
|
| 74 |
# Reindex input_df_dummies to match the columns of X_train used during training
|
| 75 |
+
input_df_aligned = input_df_dummies.reindex(columns=training_columns, fill_value=0)
|
| 76 |
|
| 77 |
#make model to predict
|
| 78 |
prediction = model.predict(input_df_aligned)
|