Spaces:
Running
Running
Update app.py
Browse files
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 =
|
43 |
batch_size: int = 32,
|
44 |
value_protection: bool = True,
|
45 |
-
rare_category_protection: bool =
|
46 |
-
flexible_generation: bool =
|
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="
|
257 |
-
placeholder="Enter a name for your
|
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=
|
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=
|
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(
|