Merge pull request #120 from 1chooo/ho-dev

update: Classifier playground
1chooo · Sep 12, 2023 · 5170867 · 5170867
2 parents aac5ff0 + be7afb7
commit 5170867
Show file tree

Hide file tree

Showing 6 changed files with 158 additions and 81 deletions.
diff --git a/Refinaid/Playground/Classifier/Dashborad/Header.py b/Refinaid/Playground/Classifier/Dashborad/Header.py
@@ -11,19 +11,28 @@
 
 class PageHeader:
 
-    def __init__(self, page_content: PageContent) -> None:
+    def __init__(
+            self, 
+            page_content: PageContent,
+        ) -> None:
         self.page_content = page_content
 
     def get_home_header(
-            self, *args: Any, **kwargs: Any) -> gr.Markdown:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> gr.Markdown:
         our_heading = gr.Markdown(
             self.page_content.home_header
         )
 
         return our_heading
 
     def get_preprocessing_header(
-            self, *args: Any, **kwargs: Any) -> gr.Markdown:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> gr.Markdown:
         our_preprocessing_header = gr.Markdown(
             self.page_content.preprocessing_header
         )

diff --git a/Refinaid/Playground/Classifier/Dashborad/Preprocessing.py b/Refinaid/Playground/Classifier/Dashborad/Preprocessing.py
@@ -12,14 +12,20 @@
 
 class PreprocessingComponent:
 
-    def __init__(self, page_content: PageContent) -> None:
+    def __init__(
+            self, 
+            page_content: PageContent,
+        ) -> None:
         self.page_content = page_content
 
     def get_dataset_info(
-            self, *args: Any, **kwargs: Any) -> Tuple[
-                gr.Markdown, 
-                gr.Dropdown
-            ]:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> Tuple[
+            gr.Markdown, 
+            gr.Dropdown
+        ]:
 
         dataset_header = gr.Markdown("### Dataset")
 
@@ -38,10 +44,13 @@ def get_dataset_info(
         )
 
     def get_select_mutiple_parameters_info(
-            self, *args: Any, **kwargs: Any) -> Tuple[
-                gr.Markdown, 
-                gr.Dropdown
-            ]:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> Tuple[
+            gr.Markdown, 
+            gr.Dropdown
+        ]:
         select_mutiple_parameters_header = gr.Markdown(
             "### Select Mutiple Parameters"
         )
@@ -71,9 +80,9 @@ def get_missing_values_handling_info(
             *args: Any, 
             **kwargs: Any,
         ) -> Tuple[
-                gr.Markdown, 
-                gr.Radio
-            ]:
+            gr.Markdown, 
+            gr.Radio
+        ]:
 
         missing_values_handling_header = gr.Markdown(
             "### Missing Values Handling"
@@ -93,11 +102,13 @@ def get_missing_values_handling_info(
         )
 
     def get_data_scale_info(
-            self, *args: Any, **kwargs: Any
+            self, 
+            *args: Any, 
+            **kwargs: Any,
         ) -> Tuple[
-                gr.Markdown, 
-                gr.Radio
-            ]:
+            gr.Markdown, 
+            gr.Radio
+        ]:
 
         data_scale_header = gr.Markdown(
             "### Data Scaling"
@@ -119,13 +130,15 @@ def get_data_scale_info(
         )
 
     def get_data_split_info(
-            self, *args: Any, **kwargs: Any
+            self, 
+            *args: Any, 
+            **kwargs: Any,
         ) -> Tuple[
-                gr.Markdown, 
-                gr.Slider, 
-                gr.Slider, 
-                gr.Slider
-            ]:
+            gr.Markdown, 
+            gr.Slider, 
+            gr.Slider, 
+            gr.Slider
+        ]:
         data_split_header = gr.Markdown(
             "### Data Split\nTotal value should be 100%"
         )
@@ -163,10 +176,12 @@ def get_data_split_info(
         )
 
     def get_submit_dataset_setting_btn(
-            self, *args: Any, **kwargs: Any
+            self, 
+            *args: Any, 
+            **kwargs: Any,
         ) -> Tuple[
-                gr.Button
-            ]:
+            gr.Button
+        ]:
 
         submit_dataset_setting_btn = gr.Button(
             value="Submit Data Preprocessing Results",
@@ -177,10 +192,13 @@ def get_submit_dataset_setting_btn(
         )
 
     def get_preprocessing_visulize_info(
-            self, *args: Any, **kwargs: Any
+            self, 
+            *args: Any, 
+            **kwargs: Any,
         ) -> Tuple[
-                gr.Markdown, gr.ScatterPlot
-            ]:
+            gr.Markdown, 
+            gr.ScatterPlot,
+        ]:
 
         preprocessing_visulize_header = gr.Markdown(
             "### Data Visualization in Preprocessing"
@@ -203,10 +221,13 @@ def get_preprocessing_visulize_info(
         )
 
     def get_preprocessing_visualize_axis_info(
-            self, *args: Any, **kwargs: Any
+            self, 
+            *args: Any, 
+            **kwargs: Any,
         ) -> Tuple[
-                gr.Dropdown, gr.Dropdown
-            ]:
+            gr.Dropdown, 
+            gr.Dropdown
+        ]:
         x_axis_dropdown = gr.Dropdown(
             label="X Axis", 
             value='PassengerId',

diff --git a/Refinaid/Playground/Classifier/Dashborad/Training.py b/Refinaid/Playground/Classifier/Dashborad/Training.py
@@ -11,13 +11,19 @@
 
 class TrainingComponent:
 
-    def __init__(self, page_content: PageContent) -> None:
+    def __init__(
+            self, 
+            page_content: PageContent,
+        ) -> None:
         self.page_content = page_content
 
     def get_training_info(
-            self, *args: Any, **kwargs: Any) -> Tuple[
-                gr.Markdown
-            ]:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> Tuple[
+            gr.Markdown
+        ]:
         training_heading = gr.Markdown(
             self.page_content.training_header    
         )
@@ -27,10 +33,13 @@ def get_training_info(
         )
 
     def get_picked_dataset_info(
-            self, *args: Any, **kwargs: Any) -> Tuple[
-                gr.Markdown, 
-                gr.Dataframe,
-            ]:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> Tuple[
+            gr.Markdown, 
+            gr.Dataframe,
+        ]:
         picked_dataset_header = gr.Markdown(
             "### Data You have picked!!!"
         )
@@ -50,9 +59,12 @@ def get_picked_dataset_info(
         )
 
     def get_model_dropdown_info(
-            self, *args: Any, **kwargs: Any) -> Tuple[
-                gr.Dropdown
-            ]:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> Tuple[
+            gr.Dropdown
+        ]:
         choices =[
             "Decision Tree Classifier", 
             "K Neighbor Classifier",
@@ -69,15 +81,18 @@ def get_model_dropdown_info(
         )
 
     def get_decision_tree_classifer_info(
-            self, *args: Any, **kwargs: Any) -> Tuple[
-                gr.Markdown,
-                gr.Dropdown,
-                gr.Textbox,
-                gr.Slider,
-                gr.Slider,
-                gr.Dropdown,
-                gr.Textbox,
-            ]:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> Tuple[
+            gr.Markdown,
+            gr.Dropdown,
+            gr.Textbox,
+            gr.Slider,
+            gr.Slider,
+            gr.Dropdown,
+            gr.Textbox,
+        ]:
         decision_tree_classifer_title = gr.Markdown(
             "### Decision Tree Classifier", 
             visible=False,
@@ -146,9 +161,15 @@ def get_decision_tree_classifer_info(
         )
 
     def get_k_neighbors_classifier_info(
-            self, *args: Any, **kwargs: Any) -> Tuple[
-                gr.Markdown, gr.Slider, gr.Dropdown, gr.Dropdown
-            ]:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> Tuple[
+            gr.Markdown, 
+            gr.Slider, 
+            gr.Dropdown, 
+            gr.Dropdown,
+        ]:
         k_neighbors_classifier_title = gr.Markdown(
             "### K Neighbors Classifier", 
             visible=False,
@@ -193,9 +214,12 @@ def get_k_neighbors_classifier_info(
         )
 
     def get_training_btn_info(
-            self, *args: Any, **kwargs: Any) -> Tuple[
-                gr.Button,
-            ]:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> Tuple[
+            gr.Button,
+        ]:
         train_btn = gr.Button(
             value="Train"
         )
@@ -205,10 +229,13 @@ def get_training_btn_info(
         )
 
     def get_training_results_info(
-            self, *args: Any, **kwargs: Any) -> Tuple[
-                gr.Markdown, 
-                gr.Dataframe,
-            ]:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> Tuple[
+            gr.Markdown, 
+            gr.Dataframe,
+        ]:
         training_results_header = gr.Markdown("## Training Result")
         training_results = gr.Dataframe(
             headers=[
@@ -228,11 +255,14 @@ def get_training_results_info(
         )
 
     def get_training_results_plot_info(
-            self, *args: Any, **kwargs: Any) -> Tuple[
-                gr.Plot, 
-                gr.Plot,
-                gr.Plot,
-            ]:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> Tuple[
+            gr.Plot, 
+            gr.Plot,
+            gr.Plot,
+        ]:
         train_img1 = gr.Plot(
             interactive=True
         )
@@ -247,4 +277,5 @@ def get_training_results_plot_info(
             train_img1,
             train_img2,
             train_img3,
-        )
+        )
+
diff --git a/Refinaid/Playground/Classifier/Dashborad/TrainingHistory.py b/Refinaid/Playground/Classifier/Dashborad/TrainingHistory.py
@@ -11,24 +11,33 @@
 
 class TrainingHistoryComponent:
 
-    def __init__(self, page_content: PageContent) -> None:
+    def __init__(
+            self, 
+            page_content: PageContent,
+        ) -> None:
         self.page_content = page_content
 
     def get_history_info(
-            self, *args: Any, **kwargs: Any) -> Tuple[
-                gr.Markdown
-            ]:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> Tuple[
+            gr.Markdown
+        ]:
         history_heading = gr.Markdown("## History")
 
         return (
             history_heading
         )
 
     def get_history_training_info(
-            self, *args: Any, **kwargs: Any) -> Tuple[
-                gr.Markdown, 
-                gr.Dataframe
-            ]:
+            self, 
+            *args: Any, 
+            **kwargs: Any,
+        ) -> Tuple[
+            gr.Markdown, 
+            gr.Dataframe
+        ]:
 
         history_heading = self.get_history_info()
         training_history = gr.Dataframe(