TabPFN commited on
Commit
a0b5459
·
1 Parent(s): 41b6283

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +1 -1
app.py CHANGED
@@ -14,6 +14,7 @@ import os
14
  import matplotlib.pyplot as plt
15
  from matplotlib.colors import ListedColormap
16
 
 
17
 
18
 
19
  def compute(df_table):
@@ -46,7 +47,6 @@ def compute(df_table):
46
  except ValueError:
47
  return "⚠️ **Please only add numbers (to the inputs) or leave fields empty.**", None, None
48
 
49
- classifier = TabPFNClassifier(base_path=tabpfn_path, device='cpu', N_ensemble_configurations=4)
50
  classifier.fit(x_train, y_train)
51
  y_eval, p_eval = classifier.predict(x_eval, return_winning_probability=True)
52
 
 
14
  import matplotlib.pyplot as plt
15
  from matplotlib.colors import ListedColormap
16
 
17
+ classifier = TabPFNClassifier(base_path=tabpfn_path, device='cpu', N_ensemble_configurations=4)
18
 
19
 
20
  def compute(df_table):
 
47
  except ValueError:
48
  return "⚠️ **Please only add numbers (to the inputs) or leave fields empty.**", None, None
49
 
 
50
  classifier.fit(x_train, y_train)
51
  y_eval, p_eval = classifier.predict(x_eval, return_winning_probability=True)
52