mertkarabacak commited on
Commit
e91efea
·
verified ·
1 Parent(s): 5562670

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -54,7 +54,7 @@ tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=8)
54
  y1_model = tabpfn
55
  y1_model = y1_model.fit(x1, y1, overwrite_warning=True)
56
 
57
- y1_calib_model = CalibratedClassifierCV(y1_model, method='sigmoid', cv='prefit')
58
  y1_calib_model = y1_calib_model.fit(x1_valid.astype(np.float64), y1_valid.astype(np.float64))
59
 
60
  y1_explainer = shap.Explainer(y1_model.predict, x1)
 
54
  y1_model = tabpfn
55
  y1_model = y1_model.fit(x1, y1, overwrite_warning=True)
56
 
57
+ y1_calib_model = CalibratedClassifierCV(y1_model, method='isotonic', cv='prefit')
58
  y1_calib_model = y1_calib_model.fit(x1_valid.astype(np.float64), y1_valid.astype(np.float64))
59
 
60
  y1_explainer = shap.Explainer(y1_model.predict, x1)