wjnwjn59 commited on
Commit
28c5210
·
1 Parent(s): 679ad4f

update parameter

Browse files
Files changed (1) hide show
  1. app.py +19 -4
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(label="Max Depth", value=5, minimum=0, maximum=20, precision=0, info="Set to 0 for unlimited depth")
302
- min_samples_split = gr.Number(label="Min Samples Split", value=2, minimum=2, maximum=100, precision=0)
303
- min_samples_leaf = gr.Number(label="Min Samples Leaf", value=1, minimum=1, maximum=50, precision=0)
 
 
 
 
 
 
 
 
 
 
 
 
304
  with gr.Row():
305
- criterion = gr.Dropdown(choices=sorted(CLASS_CRITS), value="gini", label="🎯 Criterion")
 
 
 
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: