Update my_pages/rashomon_effect.py

my_pages/rashomon_effect.py

@@ -17,7 +17,7 @@ def render():
 
     # Function to plot scatter
     def plot_scatter(x, y, colors, title="", decision_boundary=None, boundary_type=None, highlight_point=None):
-        fig, ax = plt.subplots(figsize=(5, 5))
+        fig, ax = plt.subplots(figsize=(3, 3))
         ax.scatter(x, y, c=colors, alpha=0.6)
         ax.set_xlabel("Annual Income")
         ax.set_ylabel("Credit Score")
@@ -38,6 +38,11 @@ def render():
         if highlight_point is not None:
             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.set_xticks([])
+        ax.set_yticks([])
+        
         return fig
 
     # Top scatter plot (centered to match smaller width)