Update my_pages/rashomon_effect.py

my_pages/rashomon_effect.py

@@ -45,18 +45,18 @@ def render():
         # Decision boundary
         if boundary_type is not None:
             if boundary_type == "vertical":
-                ax.axvline(65, color='blue', linestyle='--')
+                ax.axvline(65, color='blue')
                 ax.fill_betweenx(np.arange(min(y), max(y)), 65, max(x), alpha=0.1, color='green')
                 ax.fill_betweenx(np.arange(min(y), max(y)), min(x), 65, alpha=0.1, color='red')
             elif boundary_type == "horizontal":
-                ax.axhline(650, color='blue', linestyle='--')
+                ax.axhline(650, color='blue')
                 ax.fill_between(np.arange(min(x), max(x)), 650, max(y), alpha=0.1, color='green')
                 ax.fill_between(np.arange(min(x), max(x)), min(y), 650, alpha=0.1, color='red')
             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', linestyle='--')
+                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')
 
@@ -83,7 +83,7 @@ def render():
     with col3:
         st.pyplot(plot_scatter(income, credit, colors, boundary_type="slant"))
         slant_selected = st.button("Choose Model 2")
-    with col3:
+    with col4:
         st.pyplot(plot_scatter(income, credit, colors, boundary_type="horizontal"))
         horizontal_selected = st.button("Choose Model 3")