Upload app.py

app.py

@@ -24,7 +24,7 @@ from datasets import load_dataset
 
 x1 = load_dataset("mertkarabacak/DMVO-mRS", data_files="gradio_train_data.csv", use_auth_token = HF_TOKEN)
 x1 = pd.DataFrame(x1['train'])
-x1 = x1.iloc[:, 1:]
+#x1 = x1.iloc[:, 1:]
 
 print(x1.columns)
 
@@ -32,7 +32,7 @@ print(x1.columns)
 
 x1_valid = load_dataset("mertkarabacak/DMVO-mRS", data_files="gradio_validation_data.csv", use_auth_token = HF_TOKEN)
 x1_valid = pd.DataFrame(x1_valid['train'])
-x1_valid = x1_valid.iloc[:, 1:]
+#x1_valid = x1_valid.iloc[:, 1:]
 
 print(x1_valid.columns)
 
@@ -55,7 +55,7 @@ y1_model = tabpfn
 y1_model = y1_model.fit(x1, y1, overwrite_warning=True)
 
 y1_calib_model = CalibratedClassifierCV(y1_model, method='sigmoid', cv='prefit')
-y1_calib_model = y1_calib_model.fit(x1_valid, y1_valid)
+y1_calib_model = y1_calib_model.fit(x1_valid.values, y1_valid.values)
 
 y1_explainer = shap.Explainer(y1_model.predict, x1)