Debmalya commited on
Commit
aedcd41
·
1 Parent(s): 76a44ef
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -5,7 +5,7 @@ import pandas as pd
5
  import category_encoders as ce
6
  healthcare_stroke_data = pd.read_csv("healthcare-dataset-stroke-data.csv")
7
 
8
- s = setup(healthcare_stroke_data, target = 'stroke', fix_imbalance = True, session_id = 123)
9
 
10
  best = compare_models(sort = 'AUC', n_select = 15)
11
  compare_model_results = pull()
 
5
  import category_encoders as ce
6
  healthcare_stroke_data = pd.read_csv("healthcare-dataset-stroke-data.csv")
7
 
8
+ s = setup(healthcare_stroke_data, target = 'stroke', normalize = True, normalize_method = 'zscore', transformation=True, fix_imbalance = True, session_id=123, fold = 60, remove_outliers= True, outliers_threshold = 0.05, remove_multicollinearity=True, multicollinearity_threshold = 0.9)
9
 
10
  best = compare_models(sort = 'AUC', n_select = 15)
11
  compare_model_results = pull()