Update app.py

app.py

@@ -69,6 +69,7 @@ ship_type = st.selectbox("船舶種類", list(ship_type_to_code.keys()))
 
 if st.button("🔮 開始預測"):
     ship_type_encoded = ship_type_to_code[ship_type]
+    # input_df 用中文欄位名，符合模型訓練時格式
     input_df = pd.DataFrame({
         "航線組合數": [port_count],
         "年": [year],
@@ -80,25 +81,23 @@ if st.button("🔮 開始預測"):
 
     st.subheader("🧠 模型決策解釋圖（SHAP Waterfall plot）")
 
-shap_values = explainer(input_df)
+    shap_values = explainer(input_df)
 
-# 對 feature_names 改成英文
-feature_name_map = {
-    "航線組合數": "Route Count",
-    "年": "Year",
-    "月": "Month",
-    "船舶種類_編碼": "Ship Type Code"
-}
+    # 把SHAP特徵名稱轉成英文，但不改input_df欄位名稱
+    feature_name_map = {
+        "航線組合數": "Route Count",
+        "年": "Year",
+        "月": "Month",
+        "船舶種類_編碼": "Ship Type Code"
+    }
+    shap_values.feature_names = [feature_name_map.get(f, f) for f in shap_values.feature_names]
 
-shap_values.feature_names = [feature_name_map.get(f, f) for f in shap_values.feature_names]
+    ax = shap.plots.waterfall(shap_values[0], show=False)
 
-ax = shap.plots.waterfall(shap_values[0], show=False)
-
-# 負號替換
-for text in ax.texts:
-    if text.get_text().startswith('\u2212'):
-        text.set_text(text.get_text().replace('\u2212', '-'))
-
-st.pyplot(ax.figure)
-plt.close(ax.figure)
+    # 負號替換（避免顯示問題）
+    for text in ax.texts:
+        if text.get_text().startswith('\u2212'):
+            text.set_text(text.get_text().replace('\u2212', '-'))
 
+    st.pyplot(ax.figure)
+    plt.close(ax.figure)