Rename DatasetOptions to Dataset for clarity

dataset_options.py → dataset.py

@@ -38,7 +38,7 @@ def get_data_points(function, xlim=(-1, 1), nsample=10, sigma=0, seed=0):
     return x, y
 
 
-class DatasetOptions:
+class Dataset:
     def __init__(
         self,
         mode: str = "generate",
@@ -89,7 +89,7 @@ class DatasetOptions:
             raise ValueError(f"Unknown dataset mode: {self.mode}")
 
     def update(self, **kwargs):
-        return DatasetOptions(
+        return Dataset(
             mode=kwargs.get("mode", self.mode),
             function=kwargs.get("function", self.function),
             xmin=kwargs.get("xmin", self.xmin),
@@ -121,7 +121,7 @@ class DatasetOptions:
         )
 
 
-class DatasetOptionsView:
+class DatasetView:
     def update_mode(self, mode: str, state: gr.State):
         state = state.update(mode=mode)
 
@@ -183,11 +183,11 @@ class DatasetOptionsView:
             )
             with gr.Row():
                 xmin = gr.Number(
-                    label="X min", 
+                    label="x min", 
                     value=options.xmin,
                 )
                 xmax = gr.Number(
-                    label="X max", 
+                    label="x max", 
                     value=options.xmax,
                 )
             sigma = gr.Number(
@@ -215,7 +215,7 @@ class DatasetOptionsView:
             outputs=[state, function, xmin, xmax, sigma, nsample, regenerate, csv_upload],
         )
 
-        # function
+        # generate mode
         function.submit(
             lambda f, s: s.update(function=f),
             inputs=[function, state],
@@ -247,7 +247,7 @@ class DatasetOptionsView:
             outputs=[state],
         )
 
-        # csv upload
+        # csv mode
         csv_upload.upload(
             self.upload_csv,
             inputs=[csv_upload, state],

mlp_visualizer.py

@@ -29,7 +29,7 @@ logging.basicConfig(
 )
 logger = logging.getLogger("ELVIS")
 
-from dataset_options import DatasetOptions, DatasetOptionsView, get_function
+from dataset import Dataset, DatasetView, get_function
 
 
 class MlpVisualizer:
@@ -44,7 +44,7 @@ class MlpVisualizer:
     display: none;
 }"""
 
-    def plot(self, dataset_options: DatasetOptions):
+    def plot(self, dataset_options: Dataset):
         print("Plotting")
         t1 = time.time()
         fig = plt.figure(figsize=(self.canvas_width / 100., self.canvas_height / 100.0), dpi=100)
@@ -98,7 +98,7 @@ class MlpVisualizer:
             gr.HTML("<div style='text-align:left; font-size:40px; font-weight: bold;'>MLP Training Visualizer</div>")
 
             # states
-            dataset_options = gr.State(DatasetOptions())
+            dataset_options = gr.State(Dataset())
 
             # GUI elements and layout 
             with gr.Row():
@@ -111,7 +111,7 @@ class MlpVisualizer:
 
                 with gr.Column(scale=1): 
                     with gr.Tab("Dataset"):
-                        dataset_view = DatasetOptionsView()
+                        dataset_view = DatasetView()
                         dataset_view.build(state=dataset_options)
                         dataset_options.change(
                             fn=self.plot,