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

Upload app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -55,7 +55,7 @@ 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.values, y1_valid.values)
59
 
60
  y1_explainer = shap.Explainer(y1_model.predict, x1)
61
 
 
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)
61