Bjerring98 commited on
Commit
199a766
·
verified ·
1 Parent(s): 1cf5881

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -3
app.py CHANGED
@@ -476,9 +476,8 @@ elif option == "SML Classification":
476
  'Income Bracket': [income_bracket] # Directly use Income Bracket as numerical
477
  })
478
 
479
- # OneHotEncode only the categorical variables
480
- cat_encoded = pd.DataFrame(ohe.transform(cat_features).todense(),
481
- columns=ohe.get_feature_names_out(['Place of living', 'Education Level', 'Age Group']))
482
 
483
  # Combine categorical and numerical features
484
  features = pd.concat([num_features, cat_encoded], axis=1)
 
476
  'Income Bracket': [income_bracket] # Directly use Income Bracket as numerical
477
  })
478
 
479
+ cat_encoded = pd.DataFrame(ohe.transform(cat_features),
480
+ columns=ohe.get_feature_names_out(['Place of living', 'Education Level', 'Age Group']))
 
481
 
482
  # Combine categorical and numerical features
483
  features = pd.concat([num_features, cat_encoded], axis=1)