Upload 2 files

bayesian_core.py

@@ -128,7 +128,8 @@ class BayesianHierarchicalAnalyzer:
                         chains=n_chains,
                         target_accept=target_accept,
                         return_inferencedata=True,
-                        progressbar=False  # 在 Streamlit 中關閉進度條
+                        progressbar=False, # 在 Streamlit 中關閉進度條
+                        discard_tuned_samples=False  # 👈 加這行!保留 tune 樣本
                     )
                 
                 # 生成摘要統計

bayesian_utils.py

@@ -24,8 +24,21 @@ def plot_trace(trace, var_names=['d', 'sigma']):
     fig, axes = plt.subplots(len(var_names), 2, figsize=(14, 4 * len(var_names)))
     if len(var_names) == 1:
         axes = axes.reshape(1, -1)
-    
-    az.plot_trace(trace, var_names=var_names, axes=axes)
+       
+    az.plot_trace(trace, var_names=var_names, axes=axes, 
+                  combined=False,
+                  compact=False)
+
+    # 加上 Tune 結束的垂直線
+    for i in range(len(var_names)):
+        if i == 0:
+            axes[i, 1].axvline(x=1000, color='red', linestyle='--', 
+                               linewidth=2, alpha=0.7, label='Tune結束')
+            axes[i, 1].legend(loc='upper right', fontsize=9)
+        else:
+            axes[i, 1].axvline(x=1000, color='red', linestyle='--', 
+                               linewidth=2, alpha=0.7)
+
     plt.tight_layout()
     
     # 轉換為圖片