mertkarabacak commited on
Commit
71d4e94
1 Parent(s): 9833744

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -4
app.py CHANGED
@@ -102,7 +102,7 @@ y4_valid = x4_valid.pop('OUTCOME')
102
  #Training models.
103
 
104
  from tabpfn import TabPFNClassifier
105
- tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=4)
106
 
107
  y1_model = tabpfn
108
  y1_model = y1_model.fit(x1, y1, overwrite_warning=True)
@@ -114,7 +114,7 @@ y1_explainer = shap.Explainer(y1_model.predict, x1)
114
 
115
 
116
  from tabpfn import TabPFNClassifier
117
- tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=4)
118
 
119
  y2_model = tabpfn
120
  y2_model = y2_model.fit(x2, y2, overwrite_warning=True)
@@ -126,7 +126,7 @@ y2_explainer = shap.Explainer(y2_model.predict, x2)
126
 
127
 
128
  from tabpfn import TabPFNClassifier
129
- tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=4)
130
 
131
  y3_model = tabpfn
132
  y3_model = y3_model.fit(x3, y3, overwrite_warning=True)
@@ -138,7 +138,7 @@ y3_explainer = shap.Explainer(y3_model.predict, x3)
138
 
139
 
140
  from tabpfn import TabPFNClassifier
141
- tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=4)
142
 
143
  y4_model = tabpfn
144
  y4_model = y4_model.fit(x4, y4, overwrite_warning=True)
 
102
  #Training models.
103
 
104
  from tabpfn import TabPFNClassifier
105
+ tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=8)
106
 
107
  y1_model = tabpfn
108
  y1_model = y1_model.fit(x1, y1, overwrite_warning=True)
 
114
 
115
 
116
  from tabpfn import TabPFNClassifier
117
+ tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=8)
118
 
119
  y2_model = tabpfn
120
  y2_model = y2_model.fit(x2, y2, overwrite_warning=True)
 
126
 
127
 
128
  from tabpfn import TabPFNClassifier
129
+ tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=8)
130
 
131
  y3_model = tabpfn
132
  y3_model = y3_model.fit(x3, y3, overwrite_warning=True)
 
138
 
139
 
140
  from tabpfn import TabPFNClassifier
141
+ tabpfn = TabPFNClassifier(device='cpu', N_ensemble_configurations=8)
142
 
143
  y4_model = tabpfn
144
  y4_model = y4_model.fit(x4, y4, overwrite_warning=True)