Update my_pages/rashomon_effect.py

my_pages/rashomon_effect.py

@@ -26,7 +26,7 @@ def render():
     # labels = (income + credit > 100).astype(int)  # 1 = paid back, 0 = default
 
     income = np.array([80, 85, 97, 91, 78, 102, 84, 88, 45, 51, 34, 47, 38, 39, 97, 91, 38, 32])
-    credit = np.array([800, 805, 970, 910, 708, 1020, 804, 880, 450, 501, 304, 470, 380, 309, 370, 301, 1080, 902])
+    credit = np.array([970, 880, 1020, 910, 805, 800, 804, 708, 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]
@@ -55,10 +55,11 @@ def render():
             elif boundary_type == "slant":
                 slope = -10.677966       # From (94, 350) and (35, 980)
                 intercept = 1353.7288
-                y_line = slope * x + intercept
-                ax.plot(x, y_line, color='blue')
-                ax.fill_between(x, y_line, max(y), alpha=0.1, color='green')
-                ax.fill_between(x, min(y), y_line, alpha=0.1, color='red')
+                x_sorted = np.sort(x)
+                y_line = slope * x_sorted + intercept
+                ax.plot(x_sorted, y_line, color='blue')
+                ax.fill_between(x_sorted, y_line, max(y), alpha=0.1, color='green')
+                ax.fill_between(x_sorted, min(y), y_line, alpha=0.1, color='red')
 
         # Highlight specific point
         if highlight_point is not None: