mertkarabacak
commited on
Upload app.py
Browse files
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='
|
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)
|