Merge pull request #104 from DataResponsibly/feature/prepare_for_unce…

…rtainty_experiments Finalized virny demo
DataResponsibly · Jan 29, 2024 · b7188da · b7188da
2 parents 7b2806b + 4b9fd31
commit b7188da
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/virny/custom_classes/metrics_interactive_visualizer.py b/virny/custom_classes/metrics_interactive_visualizer.py
@@ -322,15 +322,15 @@ def create_web_app(self, start_app=True):
                         """)
                     fairness_metrics_vw3 = gr.Dropdown(
                         sorted(self.all_error_disparity_metrics),
-                        value=['Equalized_Odds_FPR', 'Equalized_Odds_TPR'], multiselect=True, label="Error Disparity Metrics", info="Select error disparity metrics to display on the heatmap:",
+                        value=[EQUALIZED_ODDS_FPR, EQUALIZED_ODDS_TPR], multiselect=True, label="Error Disparity Metrics", info="Select error disparity metrics to display on the heatmap:",
                     )
                     group_uncertainty_metrics_vw3 = gr.Dropdown(
                         sorted(self.all_uncertainty_disparity_metrics),
-                        value=['Aleatoric_Uncertainty_Ratio', 'Overall_Uncertainty_Parity'], multiselect=True, label="Uncertainty Disparity Metrics", info="Select uncertainty disparity metrics to display on the heatmap:",
+                        value=[ALEATORIC_UNCERTAINTY_RATIO, OVERALL_UNCERTAINTY_DIFFERENCE], multiselect=True, label="Uncertainty Disparity Metrics", info="Select uncertainty disparity metrics to display on the heatmap:",
                     )
                     group_stability_metrics_vw3 = gr.Dropdown(
                         sorted(self.all_stability_disparity_metrics),
-                        value=['Label_Stability_Ratio', 'Std_Parity'], multiselect=True, label="Stability Disparity Metrics", info="Select stability disparity metrics to display on the heatmap:",
+                        value=[LABEL_STABILITY_RATIO, STD_DIFFERENCE], multiselect=True, label="Stability Disparity Metrics", info="Select stability disparity metrics to display on the heatmap:",
                     )
             with gr.Row():
                 with gr.Column():
@@ -423,7 +423,7 @@ def create_web_app(self, start_app=True):
                     with gr.Row():
                         group_uncertainty_metrics_vw4 = gr.Dropdown(
                             sorted(self.all_uncertainty_disparity_metrics),
-                            value=ALEATORIC_UNCERTAINTY_PARITY, multiselect=False, label="Uncertainty Disparity Metric",
+                            value=ALEATORIC_UNCERTAINTY_DIFFERENCE, multiselect=False, label="Uncertainty Disparity Metric",
                             scale=2
                         )
                         group_uncertainty_min_val_vw4 = gr.Text(value="-1.0", label="Min value", scale=1)