Upload bayesian_core.py

bayesian_core.py

@@ -355,45 +355,28 @@ class BayesianHierarchicalAnalyzer:
             raise ValueError("請先執行分析")
         
         try:
-            import re
-            
-            # 使用 PyMC 原本的圖
             gv = pm.model_to_graphviz(self.model)
             
-            # 獲取欄位資訊 (動態)
-            control_prefix = getattr(self, 'col_control_win', 'control').replace('_win', '').replace('_battles', '').replace('_total', '')
-            treatment_prefix = getattr(self, 'col_treatment_win', 'treatment').replace('_win', '').replace('_battles', '').replace('_total', '')
-            control_win = getattr(self, 'col_control_win', 'control_win')
-            treatment_win = getattr(self, 'col_treatment_win', 'treatment_win')
-            
-            # 替換標籤 (只改文字,不改圖結構)
-            source = gv.source
-            
-            # 替換模型變數的標籤
-            replacements = {
-                # PyMC 變數名 -> 顯示名稱
-                f'label="{f"p_{control_prefix}"}"': f'label="p_{control_prefix}[i]\\n({control_prefix.capitalize()} Win Rate)"',
-                f'label="{f"p_{treatment_prefix}"}"': f'label="p_{treatment_prefix}[i]\\n({treatment_prefix.capitalize()} Win Rate)"',
-                f'label="{control_win}_obs"': f'label="{control_win}_obs[i]\\n({control_prefix.capitalize()} Wins)"',
-                f'label="{treatment_win}_obs"': f'label="{treatment_win}_obs[i]\\n({treatment_prefix.capitalize()} Wins)"',
-                'label="d"': f'label="d\\n(Overall Effect\\n{treatment_prefix} vs {control_prefix})"',
-                'label="tau"': 'label="tau\\n(Precision)"',
-                'label="sigma"': 'label="sigma\\n(Between-Pair SD)"',
-                'label="mu"': 'label="mu[i]\\n(Baseline)"',
-                'label="delta"': f'label="delta[i]\\n({treatment_prefix.capitalize()} Advantage)"',
-                'label="delta_new"': 'label="delta_new\\n(New Pair)"',
-                'label="or_speed"': 'label="OR\\n(Odds Ratio)"',
-            }
-            
-            # 執行替換
-            for old, new in replacements.items():
-                source = source.replace(old, new)
-            
-            # 更新 graphviz 物件
-            gv.source = source
+            # 嘗試美化標籤 (如果失敗就用原圖)
+            try:
+                # 獲取欄位前綴
+                control_prefix = self.col_control_win.replace('_win', '').replace('_battles', '').replace('_total', '')
+                treatment_prefix = self.col_treatment_win.replace('_win', '').replace('_battles', '').replace('_total', '')
+                
+                # 簡單替換標籤
+                src = gv.source
+                src = src.replace('label="d"', f'label="d\\n({treatment_prefix} vs {control_prefix})"')
+                src = src.replace(f'label="p_{control_prefix}"', f'label="p_{control_prefix}[i]"')
+                src = src.replace(f'label="p_{treatment_prefix}"', f'label="p_{treatment_prefix}[i]"')
+                src = src.replace(f'label="{self.col_control_win}_obs"', f'label="{self.col_control_win}_obs[i]"')
+                src = src.replace(f'label="{self.col_treatment_win}_obs"', f'label="{self.col_treatment_win}_obs[i]"')
+                src = src.replace('label="mu"', 'label="mu[i]"')
+                src = src.replace('label="delta"', 'label="delta[i]"')
+                gv.source = src
+            except:
+                pass
             
             return gv
-            
         except Exception as e:
             raise Exception(f"無法生成 DAG 圖: {str(e)}")