Spaces:
Sleeping
Sleeping
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(
|