ZennyKenny commited on
Commit
7618b82
·
verified ·
1 Parent(s): 6f7ec0d

Update app.py

Browse files
Files changed (1) hide show
  1. app.py +7 -7
app.py CHANGED
@@ -39,11 +39,11 @@ class SyntheticDataGenerator:
39
  data: pd.DataFrame,
40
  name: str,
41
  epochs: int = 10,
42
- max_training_time: int = 60,
43
  batch_size: int = 32,
44
  value_protection: bool = True,
45
- rare_category_protection: bool = True,
46
- flexible_generation: bool = True,
47
  model_size: str = "MEDIUM",
48
  target_accuracy: float = 0.95,
49
  validation_split: float = 0.2,
@@ -253,8 +253,8 @@ def create_interface():
253
  with gr.Column(scale=1):
254
  model_name = gr.Textbox(
255
  value="My Synthetic Model",
256
- label="Model Name",
257
- placeholder="Enter a name for your model",
258
  info="Appears in training runs and saved generators."
259
  )
260
  epochs = gr.Slider(
@@ -272,12 +272,12 @@ def create_interface():
272
  value_protection = gr.Checkbox(
273
  label="Value Protection",
274
  info="Adds protections to reduce memorization of unique or sensitive values.",
275
- value=True
276
  )
277
  rare_category_protection = gr.Checkbox(
278
  label="Rare Category Protection",
279
  info="Prevents overfitting to infrequent categories to improve privacy and robustness.",
280
- value=True
281
  )
282
  with gr.Column(scale=1):
283
  flexible_generation = gr.Checkbox(
 
39
  data: pd.DataFrame,
40
  name: str,
41
  epochs: int = 10,
42
+ max_training_time: int = 30,
43
  batch_size: int = 32,
44
  value_protection: bool = True,
45
+ rare_category_protection: bool = False,
46
+ flexible_generation: bool = False,
47
  model_size: str = "MEDIUM",
48
  target_accuracy: float = 0.95,
49
  validation_split: float = 0.2,
 
253
  with gr.Column(scale=1):
254
  model_name = gr.Textbox(
255
  value="My Synthetic Model",
256
+ label="Generator Name",
257
+ placeholder="Enter a name for your generator",
258
  info="Appears in training runs and saved generators."
259
  )
260
  epochs = gr.Slider(
 
272
  value_protection = gr.Checkbox(
273
  label="Value Protection",
274
  info="Adds protections to reduce memorization of unique or sensitive values.",
275
+ value=False
276
  )
277
  rare_category_protection = gr.Checkbox(
278
  label="Rare Category Protection",
279
  info="Prevents overfitting to infrequent categories to improve privacy and robustness.",
280
+ value=False
281
  )
282
  with gr.Column(scale=1):
283
  flexible_generation = gr.Checkbox(