Spaces:
Sleeping
Sleeping
Update app.py
Browse files
app.py
CHANGED
@@ -44,11 +44,14 @@ with col2:
|
|
44 |
if st.button("Predict"):
|
45 |
# Feature Scaling
|
46 |
numerical_cols = ['day', 'month', 'year', 'shop_id', 'item_id', 'item_price', 'item_category_id']
|
47 |
-
scaler
|
48 |
input_df = pd.DataFrame(input_data, index=[0])
|
49 |
-
input_df_scaled = scaler.
|
50 |
input_df_scaled = pd.DataFrame(input_df_scaled, columns=numerical_cols)
|
51 |
|
|
|
|
|
|
|
52 |
# Make predictions using the trained model
|
53 |
predictions = rf_model.predict(input_df_scaled)
|
54 |
|
|
|
44 |
if st.button("Predict"):
|
45 |
# Feature Scaling
|
46 |
numerical_cols = ['day', 'month', 'year', 'shop_id', 'item_id', 'item_price', 'item_category_id']
|
47 |
+
scaler.fit(input_data)
|
48 |
input_df = pd.DataFrame(input_data, index=[0])
|
49 |
+
input_df_scaled = scaler.transform(input_df[numerical_cols])
|
50 |
input_df_scaled = pd.DataFrame(input_df_scaled, columns=numerical_cols)
|
51 |
|
52 |
+
# Fit the model
|
53 |
+
rf_model.fit(input_df_scaled)
|
54 |
+
|
55 |
# Make predictions using the trained model
|
56 |
predictions = rf_model.predict(input_df_scaled)
|
57 |
|