update parameter
Browse files
app.py
CHANGED
|
@@ -298,11 +298,26 @@ with gr.Blocks(theme="gstaff/sketch", css=vlai_template.custom_css, fill_width=T
|
|
| 298 |
with gr.Accordion("⚙️ Parameters & Input", open=True):
|
| 299 |
gr.Markdown("**🌳 Decision Tree Parameters**")
|
| 300 |
with gr.Row():
|
| 301 |
-
max_depth = gr.Number(
|
| 302 |
-
|
| 303 |
-
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 304 |
with gr.Row():
|
| 305 |
-
criterion = gr.Dropdown(
|
|
|
|
|
|
|
|
|
|
| 306 |
|
| 307 |
inputs_group = gr.Group(visible=False)
|
| 308 |
with inputs_group:
|
|
|
|
| 298 |
with gr.Accordion("⚙️ Parameters & Input", open=True):
|
| 299 |
gr.Markdown("**🌳 Decision Tree Parameters**")
|
| 300 |
with gr.Row():
|
| 301 |
+
max_depth = gr.Number(
|
| 302 |
+
label="Max Depth",
|
| 303 |
+
value=5, minimum=0, maximum=20, precision=0,
|
| 304 |
+
info="Set to 0 for unlimited depth"
|
| 305 |
+
)
|
| 306 |
+
min_samples_split = gr.Number(
|
| 307 |
+
label="Min Samples Split",
|
| 308 |
+
value=2, minimum=2, maximum=100, precision=0,
|
| 309 |
+
info="Minimum samples required to split an internal node"
|
| 310 |
+
)
|
| 311 |
+
min_samples_leaf = gr.Number(
|
| 312 |
+
label="Min Samples Leaf",
|
| 313 |
+
value=1, minimum=1, maximum=50, precision=0,
|
| 314 |
+
info="Minimum samples required to be at a leaf node"
|
| 315 |
+
)
|
| 316 |
with gr.Row():
|
| 317 |
+
criterion = gr.Dropdown(
|
| 318 |
+
choices=sorted(CLASS_CRITS), value="gini", label="🎯 Criterion",
|
| 319 |
+
info="Objective to measure split quality (auto-switched for regression)"
|
| 320 |
+
)
|
| 321 |
|
| 322 |
inputs_group = gr.Group(visible=False)
|
| 323 |
with inputs_group:
|