TabPFN commited on
Commit
2083ffe
·
1 Parent(s): b0fb0fe

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +3 -1
app.py CHANGED
@@ -14,6 +14,9 @@ import os
14
  import matplotlib.pyplot as plt
15
  from matplotlib.colors import ListedColormap
16
 
 
 
 
17
  def compute(df_table):
18
  headers = df_table.columns
19
  table = df_table.to_numpy()
@@ -71,7 +74,6 @@ def compute(df_table):
71
 
72
  ax.scatter(x_train[:, 0], x_train[:, 1], c=y_train_index, cmap=cm_bright)
73
 
74
- classifier = TabPFNClassifier(base_path=tabpfn_path, device='cpu', N_ensemble_configurations=4)
75
  classifier.fit(x_train[:, 0:2], y_train)
76
 
77
  DecisionBoundaryDisplay.from_estimator(
 
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):
21
  headers = df_table.columns
22
  table = df_table.to_numpy()
 
74
 
75
  ax.scatter(x_train[:, 0], x_train[:, 1], c=y_train_index, cmap=cm_bright)
76
 
 
77
  classifier.fit(x_train[:, 0:2], y_train)
78
 
79
  DecisionBoundaryDisplay.from_estimator(