RamAnanth1 commited on
Commit
58e5dd1
1 Parent(s): cb31cda

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -5,7 +5,7 @@ from scipy.io import arff
5
 
6
  rtf_model = REaLTabFormer(
7
  model_type="tabular",
8
- epochs=50,
9
  gradient_accumulation_steps=4)
10
 
11
 
@@ -15,7 +15,7 @@ def generate_data(file, num_samples):
15
  df = pd.DataFrame(data[0])
16
  elif '.csv' in file.name:
17
  df = pd.read_csv(file.name)
18
- rtf_model.fit(df)
19
  # Generate synthetic data
20
  samples = rtf_model.sample(n_samples=num_samples)
21
 
 
5
 
6
  rtf_model = REaLTabFormer(
7
  model_type="tabular",
8
+ epochs=50, # Default is 200
9
  gradient_accumulation_steps=4)
10
 
11
 
 
15
  df = pd.DataFrame(data[0])
16
  elif '.csv' in file.name:
17
  df = pd.read_csv(file.name)
18
+ rtf_model.fit(df, num_bootstrap=100) # Default is 500
19
  # Generate synthetic data
20
  samples = rtf_model.sample(n_samples=num_samples)
21