Update my_pages/rashomon_effect.py

my_pages/rashomon_effect.py

@@ -29,8 +29,8 @@ def render():
 
     st.markdown("---")
 
-    income = np.array([80, 85, 97, 91, 78, 102, 84, 88, 45, 51, 34, 47, 38, 39, 97, 91, 38, 32])
-    credit = np.array([970, 880, 1020, 910, 805, 800, 804, 708, 470, 309, 450, 304, 380, 501, 370, 301, 1080, 902])
+    income = np.array([80, 85, 97, 91, 78, 102, 84, 88, 81, 40, 45, 51, 34, 47, 38, 39, 97, 91, 38, 32])
+    credit = np.array([970, 880, 1020, 910, 805, 800, 804, 708, 810, 370, 470, 309, 450, 304, 380, 501, 370, 301, 1080, 902])
     labels = np.array([1, 1, 1, 1, 1, 1, 1, 1, 0, 0, 0, 0, 0, 0, 1, 0, 0, 1])
 
     colors = ['green' if label == 1 else 'red' for label in labels]
@@ -67,7 +67,7 @@ def render():
 
         # Highlight specific point
         if highlight_point is not None:
-            ax.scatter(*highlight_point, c='yellow', edgecolors='black', s=200, zorder=5)
+            ax.scatter(*highlight_point, c='green', edgecolors='black', s=200, zorder=5, linewidths=3)
 
         ax.spines['right'].set_visible(False)
         ax.spines['top'].set_visible(False)
@@ -90,6 +90,7 @@ def render():
     col1, col2, col3, col4, col5 = st.columns([0.5, 1, 1, 1, 0.5])
     with col2:
         st.pyplot(plot_scatter(income, credit, colors, boundary_type="vertical", highlight_point=highlight_point))
+        st.markdown("Accuracy: 90%")
         if graph_selected=="vertical":
             button_click_v = st.button("Choose Model 1", type="primary")
         else:
@@ -100,6 +101,7 @@ def render():
             st.rerun()
     with col3:
         st.pyplot(plot_scatter(income, credit, colors, boundary_type="slant", highlight_point=highlight_point))
+        st.markdown("Accuracy: 90%")
         if graph_selected=="slant":
             button_click_s = st.button("Choose Model 2", type="primary")
         else:
@@ -110,6 +112,7 @@ def render():
             st.rerun()
     with col4:
         st.pyplot(plot_scatter(income, credit, colors, boundary_type="horizontal", highlight_point=highlight_point))
+        st.markdown("Accuracy: 90%")
         if graph_selected=="horizontal":
             button_click_h = st.button("Choose Model 3", type="primary")
         else:
@@ -119,6 +122,15 @@ def render():
             st.session_state.graph_selected = "horizontal"
             st.rerun()
 
+    if "graph_selected" in st.session_state:
+        multiplicity_message = "Notice the highlighted individual who should have gotten loan but your selected model rejected them.\
+                                They are asking justification for selecting this model, when there existed another model with same accuracy\
+                                that would have given this individual the loan!"
+        st.markdown(
+            f"<div style='text-align:center; color:#c0392b; font-size:20px; font-weight:bold; margin:14px 0;'>{multiplicity_message}</div>",
+            unsafe_allow_html=True,
+        )
+    
     st.markdown("---")
     col1, col2, col3, col4 = st.columns([2, 1, 1, 1])
     with col3: