Update my_pages/rashomon_effect.py

my_pages/rashomon_effect.py

@@ -39,24 +39,22 @@ def render():
             ax.scatter(*highlight_point, c='yellow', edgecolors='black', s=200, zorder=5)
 
         ax.spines['right'].set_visible(False)
-        ax.spines['bottom'].set_visible(False)
+        ax.spines['top'].set_visible(False)
         ax.set_xticks([])
         ax.set_yticks([])
         
         return fig
 
     # Top scatter plot (centered to match smaller width)
-    col1, col2, col3 = st.columns([1, 2, 1])
-    with col2:
+    col1, col2, col3, col4, col5 = st.columns([1, 1, 1, 1, 1])
+    with col3:
         st.pyplot(plot_scatter(income, credit, colors, title="Original Data"))
 
-    # Side-by-side decision boundary plots
-    col_left, col_right = st.columns(2)
-    with col_left:
+    with col2:
         st.pyplot(plot_scatter(income, credit, colors, title="Vertical Boundary",
                                decision_boundary=55, boundary_type="vertical"))
         left_selected = st.button("Choose Vertical Boundary")
-    with col_right:
+    with col4:
         st.pyplot(plot_scatter(income, credit, colors, title="Horizontal Boundary",
                                decision_boundary=55, boundary_type="horizontal"))
         right_selected = st.button("Choose Horizontal Boundary")
@@ -64,8 +62,7 @@ def render():
     # Show new individual based on selection
     if left_selected:
         new_point = (40, 80)  # High credit score, low income
-        col1, col2, col3 = st.columns([1, 2, 1])
-        with col2:
+        with col3:
             st.pyplot(plot_scatter(income, credit, colors,
                                    title="Vertical Boundary + New Individual",
                                    decision_boundary=55, boundary_type="vertical",
@@ -73,8 +70,7 @@ def render():
         st.warning("This individual was rejected by your chosen model. Why not choose a model that helps them?")
     elif right_selected:
         new_point = (80, 40)  # Low credit score, high income
-        col1, col2, col3 = st.columns([1, 2, 1])
-        with col2:
+        with col3:
             st.pyplot(plot_scatter(income, credit, colors,
                                    title="Horizontal Boundary + New Individual",
                                    decision_boundary=55, boundary_type="horizontal",