TabPFN commited on
Commit
3383b5d
·
1 Parent(s): d5f5e05

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -73,12 +73,12 @@ def compute(table: np.array):
73
  classifier.fit(x_train[:, 0:2], y_train)
74
 
75
  DecisionBoundaryDisplay.from_estimator(
76
- classifier, x_train[:, 0:2], alpha=0.6, ax=ax, eps=2.0, grid_resolution=100, response_method="predict_proba"
77
  )
78
  plt.xlabel(headers[0])
79
  plt.ylabel(headers[1])
80
 
81
- return None, out_table, fig
82
 
83
 
84
  def upload_file(file, remove_entries=10):
 
73
  classifier.fit(x_train[:, 0:2], y_train)
74
 
75
  DecisionBoundaryDisplay.from_estimator(
76
+ classifier, x_train[:, 0:2], alpha=0.6, ax=ax, eps=2.0, grid_resolution=25, response_method="predict_proba"
77
  )
78
  plt.xlabel(headers[0])
79
  plt.ylabel(headers[1])
80
 
81
+ return "The plot visualizes a predictor based on only two features and for two classes. The tabular results below are based on the full dataset.", out_table, fig
82
 
83
 
84
  def upload_file(file, remove_entries=10):