Spaces:

MSHS-Neurosurgery-Research
/

NCDB-GBM

Runtime error

mertkarabacak commited on May 6, 2024

Commit

7123dfc

verified ·

1 Parent(s): da25eff

Update app.py

Files changed (1) hide show

app.py CHANGED Viewed

@@ -102,7 +102,7 @@ y4_valid = x4_valid.pop('OUTCOME')
 #Training models.
 from tabpfn import TabPFNClassifier
-tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=1)
 y1_model = tabpfn
 y1_model = y1_model.fit(x1, y1, overwrite_warning=True)
@@ -114,7 +114,7 @@ y1_explainer = shap.Explainer(y1_calib_model.predict, x1)
 from tabpfn import TabPFNClassifier
-tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=1)
 y2_model = tabpfn
 y2_model = y2_model.fit(x2, y2, overwrite_warning=True)
@@ -126,7 +126,7 @@ y2_explainer = shap.Explainer(y2_calib_model.predict, x2)
 from tabpfn import TabPFNClassifier
-tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=1)
 y3_model = tabpfn
 y3_model = y3_model.fit(x3, y3, overwrite_warning=True)
@@ -138,7 +138,7 @@ y3_explainer = shap.Explainer(y3_calib_model.predict, x3)
 from tabpfn import TabPFNClassifier
-tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=1)
 y4_model = tabpfn
 y4_model = y4_model.fit(x4, y4, overwrite_warning=True)

 #Training models.
 from tabpfn import TabPFNClassifier
+tabpfn = TabPFNClassifier(device='cuda', N_ensemble_configurations=1)
 y1_model = tabpfn
 y1_model = y1_model.fit(x1, y1, overwrite_warning=True)
 from tabpfn import TabPFNClassifier
+tabpfn = TabPFNClassifier(device='cuda', N_ensemble_configurations=1)
 y2_model = tabpfn
 y2_model = y2_model.fit(x2, y2, overwrite_warning=True)
 from tabpfn import TabPFNClassifier
+tabpfn = TabPFNClassifier(device='cuda', N_ensemble_configurations=1)
 y3_model = tabpfn
 y3_model = y3_model.fit(x3, y3, overwrite_warning=True)
 from tabpfn import TabPFNClassifier
+tabpfn = TabPFNClassifier(device='cuda', N_ensemble_configurations=1)
 y4_model = tabpfn
 y4_model = y4_model.fit(x4, y4, overwrite_warning=True)