2024/03/09/18:30

app.py

@@ -2002,7 +2002,7 @@ def choose_custom_dataset(file: str):
     return get_return(True, {choose_custom_dataset_file: gr.File(Dataset.file, visible=True)})
 
 
-def select_model_optimize_radio(optimize="无"):
+def select_model_optimize_radio(optimize):
     optimize = Dataset.get_optimize_name_mapping()[optimize]
 
     if optimize == "grid_search":
@@ -2015,6 +2015,19 @@ def select_model_optimize_radio(optimize="无"):
     return get_return(True)
 
 
+def reset_select_model_optimize_radio_part_1():
+    Dataset.choose_optimize = "None"
+
+    return get_return(True)
+
+
+def reset_select_model_optimize_radio_part_2():
+    # TODO: 默认只有“网格搜索”
+    Dataset.choose_optimize = "bayes_search"
+
+    return get_return(True)
+
+
 def linear_regression_model_radio_change(model):
     if model:
         Dataset.linear_regression_model_type = Dataset.get_linear_regression_model_name_mapping()[model]
@@ -2432,13 +2445,16 @@ with gr.Blocks(js="./design/welcome.js", css="./design/custom.css") as demo:
     )
 
     # 数据模型
+    # TODO: 有BUG
     select_as_model_radio.change(
         fn=select_as_model,
         inputs=[select_as_model_radio],
         outputs=get_outputs()
     ).then(
-        fn=select_model_optimize_radio,
-        inputs=[model_optimize_radio],
+        fn=reset_select_model_optimize_radio_part_1(),
+        outputs=get_outputs()
+    ).then(
+        fn=reset_select_model_optimize_radio_part_2(),
         outputs=get_outputs()
     )