rasmodev commited on
Commit
f767b5a
·
verified ·
1 Parent(s): 1ae3560

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +5 -2
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 = scaler
48
  input_df = pd.DataFrame(input_data, index=[0])
49
- input_df_scaled = scaler.fit_transform(input_df[numerical_cols])
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