mohramzan commited on
Commit
362fed2
1 Parent(s): f5e2d5e

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -82,8 +82,8 @@ def feature_importance(age, job, marital, education, default, housing,
82
  df_transformed.columns = dv.get_feature_names_out()
83
  important_features = pd.DataFrame({'cols':df_transformed.columns, 'imp':model.feature_importances_}
84
  ).sort_values('imp', ascending=False)
85
- return important_features.plot('cols', 'imp', 'barh', figsize=(12,7), legend=False)
86
- # return print(important_features.to_json())
87
 
88
 
89
 
82
  df_transformed.columns = dv.get_feature_names_out()
83
  important_features = pd.DataFrame({'cols':df_transformed.columns, 'imp':model.feature_importances_}
84
  ).sort_values('imp', ascending=False)
85
+ #return important_features.plot('cols', 'imp', 'barh', figsize=(12,7), legend=False)
86
+ return print(important_features.to_json())
87
 
88
 
89