RamAnanth1 commited on
Commit
f68b438
1 Parent(s): 390d8ba

Reduce number of bootstraps for faster inference for demo

Browse files
Files changed (1) hide show
  1. app.py +2 -2
app.py CHANGED
@@ -16,7 +16,7 @@ def generate_data(file, num_samples):
16
  df = pd.DataFrame(data[0])
17
  elif '.csv' in file.name:
18
  df = pd.read_csv(file.name)
19
- rtf_model.fit(df, num_bootstrap=100) # Default is 500
20
  # Generate synthetic data
21
  samples = rtf_model.sample(n_samples=num_samples)
22
 
@@ -53,7 +53,7 @@ def generate_relational_data(parent_file, child_file, join_on):
53
  df=child_df,
54
  in_df=parent_df,
55
  join_on=join_on,
56
- num_bootstrap=100)
57
 
58
  # Generate parent samples.
59
  parent_samples = rtf_model.sample(5)
 
16
  df = pd.DataFrame(data[0])
17
  elif '.csv' in file.name:
18
  df = pd.read_csv(file.name)
19
+ rtf_model.fit(df, num_bootstrap=10) # Default is 500
20
  # Generate synthetic data
21
  samples = rtf_model.sample(n_samples=num_samples)
22
 
 
53
  df=child_df,
54
  in_df=parent_df,
55
  join_on=join_on,
56
+ num_bootstrap=10)
57
 
58
  # Generate parent samples.
59
  parent_samples = rtf_model.sample(5)