Hugging Face's logo

Spaces:
MSHS-Neurosurgery-Research
/
NSQIP-ACC
Sleeping

App Files Community
mertkarabacak commited on
Commit
bfaadf3
·
verified ·
1 Parent(s): 541c63a

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -49,7 +49,7 @@ y1_valid = x1_valid.pop('OUTCOME')
49
 
50
  #Training model.
51
  from tabpfn import TabPFNClassifier
52
- tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=8)
53
 
54
  y1_model = tabpfn
55
  y1_model = y1_model.fit(x1, y1, overwrite_warning=True)
 
49
 
50
  #Training model.
51
  from tabpfn import TabPFNClassifier
52
+ tabpfn = TabPFNClassifier(device='cuda', N_ensemble_configurations=8)
53
 
54
  y1_model = tabpfn
55
  y1_model = y1_model.fit(x1, y1, overwrite_warning=True)