Hugging Face's logo

Spaces:
MSHS-Neurosurgery-Research
/
NSQIP-ACC
Sleeping

App Files Community
mertkarabacak commited on
Commit
f919cf9
·
verified ·
1 Parent(s): a7c2524

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -39,7 +39,7 @@ y1 = x1.pop('OUTCOME')
39
  from tabpfn import TabPFNClassifier
40
  tabpfn = TabPFNClassifier(device='cuda', N_ensemble_configurations=8)
41
 
42
- y1_calib_model = CalibratedClassifierCV(tabpfn, method='sigmoid', cv=5)
43
  y1_calib_model = y1_calib_model.fit(x1, y1)
44
 
45
  y1_explainer = shap.Explainer(y1_model.predict, x1)
 
39
  from tabpfn import TabPFNClassifier
40
  tabpfn = TabPFNClassifier(device='cuda', N_ensemble_configurations=8)
41
 
42
+ y1_calib_model = CalibratedClassifierCV(tabpfn, method='isotonic', cv=5)
43
  y1_calib_model = y1_calib_model.fit(x1, y1)
44
 
45
  y1_explainer = shap.Explainer(y1_model.predict, x1)