Add automatic settings generation

backend/src/logic.py

@@ -260,4 +260,27 @@ def compute_plot_values(
         reg_levels=reg_levels,
         unreg_solution=unreg_solution,
         path=path,
-    )
+    )
+
+
+def compute_suggested_settings(
+    dataset: Dataset
+) -> tuple[tuple[float, float], tuple[float, float], list[float]]:
+    x = np.stack([dataset.x1, dataset.x2], axis=1)
+    moore_penrose = np.linalg.pinv(x) @ np.array(dataset.y)
+
+    if np.isclose(moore_penrose, 0).all():
+        w1_range = (-10, 10)
+        w2_range = (-10, 10)
+        return w1_range, w2_range, []
+
+    width = np.max(np.abs(moore_penrose)) * 3
+
+    w1_range = (-width, width)
+    w2_range = (-width, width)
+
+    opt_norm = float(np.linalg.norm(moore_penrose, ord=2))
+
+    reg_levels = [i / 4 * opt_norm for i in range(3)]
+
+    return w1_range, w2_range, reg_levels

backend/src/manager.py

@@ -11,6 +11,7 @@ from logic import (
     Dataset,
     PlotsData,
     compute_plot_values,
+    compute_suggested_settings,
     generate_dataset,
     load_dataset_from_csv,
 )
@@ -156,6 +157,18 @@ class Manager:
         strength_plot = self._generate_strength_plot(self.plots_data.path)
         return self, contour_plot, data_plot, strength_plot
 
+    def handle_use_suggested_settings(self) -> tuple[Manager, str, str, str]:
+        if self.dataset is None:
+            raise ValueError("Dataset is not initialized")
+
+        w1_range, w2_range, reg_levels = compute_suggested_settings(self.dataset)
+
+        w1_range_input = f"{w1_range[0]:.2f}, {w1_range[1]:.2f}"
+        w2_range_input = f"{w2_range[0]:.2f}, {w2_range[1]:.2f}"
+        reg_levels_input = ", ".join(f"{level:.2f}" for level in reg_levels)
+
+        return self, w1_range_input, w2_range_input, reg_levels_input
+
     @staticmethod
     def _generate_contour_plot(plots_data: PlotsData) -> Figure:
         fig, ax = plt.subplots(figsize=(8, 8))

frontends/gradio/main.py

@@ -92,6 +92,11 @@ def handle_generate_plots(
         resolution,
     )
 
+def handle_use_suggested_settings(
+    manager: Manager,
+) -> tuple[Manager, str, str, str]:
+    return manager.handle_use_suggested_settings()
+
 
 def launch():
     default_dataset_type = "Generate"
@@ -256,14 +261,14 @@ def launch():
                             value=default_loss_type,
                             interactive=True,
                         )
-
-                    with gr.Row():
                         regularizer_type_dropdown = gr.Dropdown(
                             label="Regularizer type",
                             choices=["l1", "l2"],
                             value=default_regularizer_type,
                             interactive=True,
                         )
+
+                    with gr.Row():
                         regularizer_levels_textbox = gr.Textbox(
                             label="Regularization levels (comma-separated)",
                             value=default_reg_levels,
@@ -290,6 +295,18 @@ def launch():
                         step=1,
                         interactive=True,
                     )
+                    
+                    gr.Button("Use suggested settings").click(
+                        fn=handle_use_suggested_settings,
+                        inputs=[manager_state],
+                        outputs=[
+                            manager_state,
+                            w1_range_textbox,
+                            w2_range_textbox,
+                            regularizer_levels_textbox,
+                        ],
+                    )
+
 
                 gr.Button("Regenerate Plots").click(
                     fn=handle_generate_plots,