Update my_pages/rashomon_effect.py

my_pages/rashomon_effect.py

@@ -25,9 +25,9 @@ def render():
     # 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]
+    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])
+    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]