Upload app.py

app.py

@@ -119,7 +119,7 @@ tabpfn = TabPFNClassifier(device='cuda', N_ensemble_configurations=1)
 y3_model = tabpfn
 y3_model = y3_model.fit(x3, y3, overwrite_warning=True)
 
-y3_calib_model = CalibratedClassifierCV(y3_model, method='sigmoid', cv='prefit')
+y3_calib_model = CalibratedClassifierCV(y3_model, method='isotonic', cv='prefit')
 y3_calib_model = y3_calib_model.fit(x3, y3)
 
 y3_explainer = shap.Explainer(y3_model.predict, x3)
@@ -131,7 +131,7 @@ tabpfn = TabPFNClassifier(device='cuda', N_ensemble_configurations=1)
 y4_model = tabpfn
 y4_model = y4_model.fit(x4, y4, overwrite_warning=True)
 
-y4_calib_model = CalibratedClassifierCV(y4_model, method='sigmoid', cv='prefit')
+y4_calib_model = CalibratedClassifierCV(y4_model, method='isotonic', cv='prefit')
 y4_calib_model = y4_calib_model.fit(x4, y4)
 
 y4_explainer = shap.Explainer(y4_model.predict, x4)
@@ -326,7 +326,7 @@ with gr.Blocks(title = "NCDB-Meningioma") as demo:
           </tr>
           <tr>
             <td>6-Month Mortality</td>
-            <td>TabPFN</td>
+            <td>Random Forest</td>
             <td>0.755 (0.733 - 0.777)</td>
             <td>0.767 (0.745 - 0.789)</td>
             <td>0.764 (0.742 - 0.786)</td>
@@ -336,7 +336,7 @@ with gr.Blocks(title = "NCDB-Meningioma") as demo:
           </tr>
           <tr>
             <td>12-Month Mortality</td>
-            <td>TabPFN</td>
+            <td>Random Forest</td>
             <td>0.685 (0.661 - 0.709)</td>
             <td>0.728 (0.705 - 0.751)</td>
             <td>0.707 (0.683 - 0.731)</td>
@@ -393,7 +393,7 @@ with gr.Blocks(title = "NCDB-Meningioma") as demo:
             
             Tumor_Size = gr.Slider(label = "Tumor Size (mm)", minimum = 1, maximum = 300, step = 1, value = 30)
             
-            Extent_of_Resection = gr.Dropdown(label = 'Extent of Resection', choices = ['No resective surgery was performed', 'Gross total resection'], type = 'index', value = 'Gross total resection')
+            Extent_of_Resection = gr.Dropdown(label = 'Extent of Resection', choices = ['No resective surgery was performed', 'Gross total resection', 'Subtotal resection'], type = 'index', value = 'Gross total resection')
             
             Radiotherapy = gr.Dropdown(label = 'Radiotherapy', choices = ['No', 'Yes'], type = 'index', value = 'Yes')