Upload mcnemar_utils.py

mcnemar_utils.py

@@ -56,7 +56,7 @@ def plot_contingency_table_heatmap(ct_labeled, feature_label, title="列聯表
 
 def plot_odds_ratio_forest(or_value, ci_low, ci_high, feature_label):
     """
-    繪製勝算比森林圖
+    繪製勝算比森林圖（線性刻度版本）
     
     Args:
         or_value: 勝算比
@@ -69,76 +69,107 @@ def plot_odds_ratio_forest(or_value, ci_low, ci_high, feature_label):
     """
     fig = go.Figure()
     
+    # 動態計算 x 軸範圍
+    x_max = max(ci_high * 1.2, 3.5)  # 至少到 3.5
+    x_min = 0
+    
     # 參考線 (OR = 1)
     fig.add_shape(
         type="line",
         x0=1, x1=1,
-        y0=-0.5, y1=0.5,
+        y0=-0.3, y1=0.3,
         line=dict(color="red", width=2, dash="dash"),
     )
     
-    # 信賴區間
+    # 信賴區間（水平線）
     fig.add_trace(go.Scatter(
         x=[ci_low, ci_high],
         y=[0, 0],
         mode='lines',
-        line=dict(color='#2d6ca2', width=3),
+        line=dict(color='#2d6ca2', width=2),
         showlegend=False,
-        hovertemplate='95% CI: [%{x:.3f}]<extra></extra>'
+        hoverinfo='skip'
     ))
     
-    # 點估計
+    # 信賴區間端點（小豎線）
+    for x_pos in [ci_low, ci_high]:
+        fig.add_shape(
+            type="line",
+            x0=x_pos, x1=x_pos,
+            y0=-0.05, y1=0.05,
+            line=dict(color='#2d6ca2', width=2)
+        )
+    
+    # 點估計（藍色圓點）
     fig.add_trace(go.Scatter(
         x=[or_value],
         y=[0],
         mode='markers',
         marker=dict(
-            size=15,
-            color='#d62728',
-            line=dict(color='white', width=2)
+            size=12,
+            color='#1f77b4',
+            line=dict(color='white', width=1.5)
         ),
         showlegend=False,
-        hovertemplate=f'OR: {or_value:.3f}<extra></extra>'
+        hovertemplate=f'<b>OR: {or_value:.3f}</b><br>95% CI: [{ci_low:.3f}, {ci_high:.3f}]<extra></extra>'
     ))
     
-    # 添加數值標註
+    # 添加數值標註框（右上角）
     fig.add_annotation(
-        x=or_value,
-        y=0.15,
+        x=ci_high + (x_max - ci_high) * 0.3,
+        y=0,
         text=f"OR = {or_value:.3f}<br>95% CI [{ci_low:.3f}, {ci_high:.3f}]",
-        showarrow=False,
-        font=dict(size=12, color='#1b4f72'),
-        bgcolor='rgba(255,255,255,0.8)',
+        showarrow=True,
+        arrowhead=2,
+        arrowsize=1,
+        arrowwidth=1.5,
+        arrowcolor='#2d6ca2',
+        ax=50,
+        ay=0,
+        font=dict(size=11, color='#1b4f72'),
+        bgcolor='rgba(255,255,255,0.95)',
         bordercolor='#2d6ca2',
-        borderwidth=1,
-        borderpad=4
+        borderwidth=1.5,
+        borderpad=6
     )
     
     fig.update_layout(
-        title=f'勝算比 (Odds Ratio)<br><sub>{feature_label}</sub>',
+        title=dict(
+            text=f'勝算比 (Odds Ratio)<br><sub style="font-size:11px;">{feature_label}</sub>',
+            x=0.5,
+            xanchor='center'
+        ),
         xaxis_title='Odds Ratio',
         yaxis=dict(
             showticklabels=False,
             showgrid=False,
-            zeroline=False
+            zeroline=False,
+            range=[-0.35, 0.35]
         ),
-        width=700,
-        height=300,
+        width=800,
+        height=280,
         template='plotly_white',
-        #xaxis=dict(type='log', showgrid=True),
         xaxis=dict(
-            type='linear',  # 改成線性
+            type='linear',  # 改用線性刻度
             showgrid=True,
+            gridcolor='rgba(200,200,200,0.3)',
             tickmode='linear',
             tick0=0,
             dtick=0.5,  # 每 0.5 一個刻度
-            range=[0, max(ci_high * 1.2, 5)]  # 動態範圍
+            range=[x_min, x_max],
+            showline=True,
+            linewidth=1,
+            linecolor='black',
+            mirror=True
         ),
-        hovermode='closest'
+        hovermode='closest',
+        plot_bgcolor='white',
+        margin=dict(l=60, r=60, t=80, b=60)
     )
     
     return fig
 
+
 def plot_discordant_pairs(b, c, label_pos, label_neg):
     """
     繪製不一致配對比較圖