Add button to sample from GP

gp_visualizer.py

@@ -144,6 +144,8 @@ class GPVisualizer:
             "show_predictions": True,
         }
 
+        self.num_y_samples = 0
+
 
     def __init__(self, width, height):
         self.canvas_width = width
@@ -222,6 +224,11 @@ class GPVisualizer:
                 color=self.plot_cmap(3)
             )
 
+        for i in range(self.num_y_samples):
+            y_sample = self.model.sample_y(x_test, random_state=i).flatten()
+            plt.plot(x_test.flatten(), y_sample, linestyle=":", label=f"sample {i}", color=self.plot_cmap(4))
+
+
         plt.legend()
 
         buf = io.BytesIO()
@@ -301,6 +308,14 @@ class GPVisualizer:
                 self.plot_options[key] = value
         return self.plot()
 
+    def add_y_sample(self):
+        self.num_y_samples += 1
+        return self.plot()
+
+    def clear_y_samples(self):
+        self.num_y_samples = 0
+        return self.plot()
+
     def launch(self):
         # build the Gradio interface
         with gr.Blocks(css=self.css) as demo:
@@ -378,6 +393,10 @@ class GPVisualizer:
                                 show_predictions = gr.Checkbox(label="Show mean prediction", value=True)
                                 show_confidence_interval = gr.Checkbox(label="Show confidence interval", value=True)
 
+                        # sampling from GP
+                        sample_button = gr.Button("Sample from GP")
+                        clear_samples_button = gr.Button("Clear samples from GP")
+
                         #gr.Markdown(''.join(open('kernel_examples.md', 'r').readlines()))
 
                     with gr.Tab("Export"):
@@ -456,6 +475,17 @@ class GPVisualizer:
                         outputs=[self.canvas],
                     )
 
+                    # sampling from GP
+                    sample_button.click(
+                        fn=self.add_y_sample,
+                        outputs=[self.canvas],
+                    )
+                    clear_samples_button.click(
+                        fn=self.clear_y_samples,
+                        outputs=[self.canvas],
+                    )
+
+
         demo.launch()
 
 visualizer = GPVisualizer(width=1200, height=900)