Add option to change the embedder arguments

decision_boundary.py

@@ -148,6 +148,7 @@ class InteractiveDecisionBoundary:
 
         # data embedding and preprocessing values
         self.embedder_class = CoordinateProjection
+        self.embedder_args = ""
         self.normalizer_class = None
         self.jitter_std = 0
 
@@ -210,7 +211,7 @@ class InteractiveDecisionBoundary:
         return features
 
     def _embed_features(self, features, return_embedder=False):
-        embedder = self.embedder_class(n_components=2)
+        embedder = self.embedder_class(n_components=2, **parse_param_string(self.embedder_args))
         features = embedder.fit_transform(features)
 
         if return_embedder:
@@ -222,6 +223,7 @@ class InteractiveDecisionBoundary:
         self.normalizer_class = None
         self.jitter_std = 0
         self.embedder_class = CoordinateProjection
+        self.embedder_args = ""
 
     def plot(self, decision_boundary=False):
         '''
@@ -475,6 +477,9 @@ class InteractiveDecisionBoundary:
         else:
             model_params_text = "".join([f"\n\t\t{k}={repr(v)}," for k, v in model_params.items()]) + "\n\t"
 
+        embedder_args = {"n_components": 2, **parse_param_string(self.embedder_args)}
+        embedder_args_text = "".join([f"\n\t\t{k}={repr(v)}," for k, v in embedder_args.items()]) + "\n\t"
+
         template = Template(Path(self.EXPORT_CODE_TEMPLATE).read_text())
         variables = {
             'model_import_statement': model_import_stmt,
@@ -482,6 +487,7 @@ class InteractiveDecisionBoundary:
             "normalizer_import_statement": normalizer_import_stmt,
             'dataset_file': self.DATASET_FILE,
             'embedder_class': embedder_class,
+            'embedder_args': embedder_args_text,
             'model_class': model_class,
             'model_params': model_params_text,
             'fig_width': self.canvas_width / 100,
@@ -512,6 +518,11 @@ class InteractiveDecisionBoundary:
         print('updated Embedder:', self.embedder_class)
         return self.plot()
 
+    def update_embedder_args(self, embedder_args):
+        self.embedder_args = embedder_args
+        print('updated Embedder args:', self.embedder_args)
+        return self.plot()
+
     def update_normalizer(self, normalizer):
         self.normalizer_class = self.normalizers[normalizer]
         print('updated Normalizer:', self.normalizer_class)
@@ -533,15 +544,15 @@ class InteractiveDecisionBoundary:
         if type == 'Draw2D':
             self.custom_selected = True
             self.dataset_type = "Draw2D"
-            new_fields = gr.File(visible=False), gr.Dropdown(visible=False, value="None"), gr.Dropdown(visible=False, value="None"), gr.Dropdown(visible=False, value=0), gr.Dropdown(visible=False, value="CoordinateProjection"), gr.Textbox(visible=True), gr.Button(visible=True), gr.Button(visible=False)
+            new_fields = gr.File(visible=False), gr.Dropdown(visible=False, value="None"), gr.Dropdown(visible=False, value="None"), gr.Dropdown(visible=False, value=0), gr.Dropdown(visible=False, value="CoordinateProjection"), gr.Textbox(visible=False), gr.Textbox(visible=True), gr.Button(visible=True), gr.Button(visible=False)
         elif type == 'Upload':
             self.dataset_type = "Upload"
             self.custom_selected = False
-            new_fields = gr.File(visible=True), gr.Dropdown(visible=False, value="None"), gr.Dropdown(visible=True, value="None"), gr.Dropdown(visible=True, value=0), gr.Dropdown(visible=True, value="CoordinateProjection"), gr.Textbox(visible=False), gr.Button(visible=False), gr.Button(visible=False)
+            new_fields = gr.File(visible=True), gr.Dropdown(visible=False, value="None"), gr.Dropdown(visible=True, value="None"), gr.Dropdown(visible=True, value=0), gr.Dropdown(visible=True, value="CoordinateProjection"), gr.Textbox(visible=True), gr.Textbox(visible=False), gr.Button(visible=False), gr.Button(visible=False)
         elif type == 'sklearn':
             self.dataset_type = "sklearn"
             self.custom_selected = False
-            new_fields = gr.File(visible=False), gr.Dropdown(visible=True, value="None"), gr.Dropdown(visible=True, value="None"), gr.Dropdown(visible=True, value=0), gr.Dropdown(visible=True, value="CoordinateProjection"), gr.Textbox(visible=False), gr.Button(visible=False), gr.Button(visible=False)
+            new_fields = gr.File(visible=False), gr.Dropdown(visible=True, value="None"), gr.Dropdown(visible=True, value="None"), gr.Dropdown(visible=True, value=0), gr.Dropdown(visible=True, value="CoordinateProjection"), gr.Textbox(visible=True), gr.Textbox(visible=False), gr.Button(visible=False), gr.Button(visible=False)
         else:
             # TODO: better error handling
             print('Error - unknown dataset type:', type)
@@ -659,6 +670,9 @@ class InteractiveDecisionBoundary:
                         )
                         self.embedder_selector = embedder_selector
 
+                        embedder_args_textbox = gr.Textbox(label="Embedder arguments", visible=False)
+                        self.embedder_args_textbox = embedder_args_textbox
+
                         # custom data
                         label = gr.Radio(["Red", "Green", "Blue"], value="Red", label="Choose point label", visible=True, elem_id="rowheight")
                         self.label = label
@@ -725,7 +739,7 @@ class InteractiveDecisionBoundary:
                 fn=self.handle_dataset_radio,
                 inputs=dataset_radio, 
                 outputs=(
-                    self.data_image, data_table, file_chooser, sklearn_data_selector, normalizer_selector, jittering_slider, embedder_selector, label, btn_clear, btn_save
+                    self.data_image, data_table, file_chooser, sklearn_data_selector, normalizer_selector, jittering_slider, embedder_selector, embedder_args_textbox, label, btn_clear, btn_save
                 ),
             )
 
@@ -747,6 +761,12 @@ class InteractiveDecisionBoundary:
                                      inputs=embedder_selector,
                                      outputs=self.data_image)
 
+            embedder_args_textbox.change(
+                fn=self.update_embedder_args,
+                inputs=embedder_args_textbox,
+                outputs=self.data_image,
+            )
+
             normalizer_selector.change(
                 fn=self.update_normalizer,
                 inputs=normalizer_selector,

export_code_template.py.j2

@@ -84,7 +84,8 @@ def main():
     # if you want to load a model
     # model = load_model("model.pkl") 
 
-    embedder = {{ embedder_class }}(n_components=2).fit(X)
+    embedder = {{ embedder_class }}({{ embedder_args }})
+    embedder.fit(X)
     create_plot(X, y, model, embedder)
     plt.show()
     # if you want to save as image