Spaces:
Runtime error
Runtime error
Commit
·
58e5dd1
1
Parent(s):
cb31cda
Update app.py
Browse files
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 |
|