Update my_pages/rashomon_effect.py

my_pages/rashomon_effect.py

@@ -3,12 +3,14 @@ import matplotlib.pyplot as plt
 import numpy as np
 from utils import go_to
 
+plt.style.use('dark_background')
+
 def render():
     st.markdown(
         """
         <div style='text-align: center; font-size:18px; color:gray;'>
-        Consider data about individuals who paid back their loans (green) and those who defaulted (red). <br>
-        Which model out of the two will you choose as your final model to give loan applications? <br><br>
+        Consider data about individuals who either paid their loans (green) or defaulted (red). <br>
+        Which model out of the two will you choose to give loan applications? <br><br>
         </div>
         """,
         unsafe_allow_html=True
@@ -17,17 +19,21 @@ def render():
     st.markdown("---")
 
     # Generate synthetic data
-    np.random.seed(42)
-    n_points = 100
-    income = np.random.normal(50, 15, n_points)
-    credit = np.random.normal(50, 15, n_points)
-    labels = (income + credit > 100).astype(int)  # 1 = paid back, 0 = default
+    # np.random.seed(42)
+    # n_points = 100
+    # income = np.random.normal(50, 15, n_points)
+    # credit = np.random.normal(50, 15, n_points)
+    # labels = (income + credit > 100).astype(int)  # 1 = paid back, 0 = default
+
+    income = [80, 85, 97, 91, 78, 102, 84, 88, 45, 51, 34, 47, 38, 39, 97, 91, 38, 32]
+    credit = [800, 805, 970, 910, 708, 1020, 804, 880, 450, 501, 304, 470, 380, 309, 370, 301, 1080, 902]
+    labels = [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]
 
     # 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=(3, 3))
+        fig, ax = plt.subplots(figsize=(2, 2))
         ax.scatter(x, y, c=colors, alpha=0.6)
         ax.set_xlabel("Annual Income")
         ax.set_ylabel("Credit Score")