TabPFN commited on
Commit
292ba65
1 Parent(s): b392eb0

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +4 -1
app.py CHANGED
@@ -45,7 +45,10 @@ def compute(df_table):
45
  x_train = torch.tensor(np.delete(train_table, y_column, axis=1).astype(np.float32))
46
  x_eval = torch.tensor(np.delete(eval_table, y_column, axis=1).astype(np.float32))
47
 
48
- y_train = train_table[:, y_column]
 
 
 
49
  except ValueError:
50
  return "⚠️ **Please only add numbers (to the inputs) or leave fields empty.**", None, None
51
 
 
45
  x_train = torch.tensor(np.delete(train_table, y_column, axis=1).astype(np.float32))
46
  x_eval = torch.tensor(np.delete(eval_table, y_column, axis=1).astype(np.float32))
47
 
48
+ #y_train = train_table[:, y_column]
49
+ y_train = np.array(train_table[:, y_column].tolist())
50
+
51
+
52
  except ValueError:
53
  return "⚠️ **Please only add numbers (to the inputs) or leave fields empty.**", None, None
54