Upload xgb_model, requirements, website.py

requirements.txt

@@ -0,0 +1,6 @@
+xgboost==2.1.4
+numpy==1.26.4
+pandas==2.2.2
+shap==0.46.0
+gradio==5.20.1
+matplotlib==3.9.2

website.py

@@ -0,0 +1,73 @@
+import numpy as np
+import pandas as pd
+import xgboost as xgb
+import shap
+import gradio as gr
+import matplotlib.pyplot as plt
+import matplotlib
+
+matplotlib.use("Agg")  # 防止服务器无图形界面时报错
+
+# 指定输入特征顺序
+feature_names = ["CT value(HU)", "Tumor size(cm)", "ctDNA", "CEA", "Location", "CYFRA21-1", "CA125", "LDH"]
+
+# 加载模型（确保是用 sklearn API 训练并保存的）
+model = xgb.XGBClassifier()
+model.load_model("./xgb_model.json")
+model.get_booster().feature_names = feature_names
+
+# 初始化 SHAP 解释器
+explainer = shap.Explainer(model)
+
+
+# 预测函数
+def predict_probability(CT_value, Tumor_size, ctDNA, CEA, Location, CYFRA21_1, CA125, LDH):
+    input_data = pd.DataFrame(
+        [[CT_value, Tumor_size, ctDNA, CEA, Location, CYFRA21_1, CA125, LDH]], columns=feature_names
+    )
+
+    # 将 Location 和 ctDNA 转换为数值型
+    input_data["Location"] = input_data["Location"].map({"Central": 1, "Peripheral": 0})
+    input_data["ctDNA"] = input_data["ctDNA"].map({"Positive": 1, "Negative": 0})
+    # 预测
+    try:
+        prob = model.predict_proba(input_data)[0][1]
+    except Exception as e:
+        return f"预测出错: {e}", None
+
+    # 计算 SHAP 值
+    try:
+        shap_values = explainer(input_data)
+
+        # 绘图
+        shap.plots.waterfall(shap_values[0], show=False)
+        plt.title("SHAP Waterfall Plot")
+        plt.savefig("shap_plot.png", bbox_inches="tight", dpi=300)
+        plt.close()
+    except Exception as e:
+        return f"SHAP 图生成失败: {e}", None
+
+    return f"阳性概率: {prob:.2%}", "shap_plot.png"
+
+
+demo = gr.Interface(
+    fn=predict_probability,
+    inputs=[
+        gr.Number(label="CT value(HU)"),
+        gr.Number(label="Tumor size(cm)"),
+        gr.Dropdown(choices=["Positive", "Negative"], label="ctDNA"),
+        gr.Number(label="CEA (ng/mL) Normal range: 0-5"),
+        gr.Dropdown(choices=["Central", "Peripheral"], label="Location"),  # 修改为 Dropdown 类型
+        gr.Number(label="CYFRA21-1 (ng/mL) Normal range: 0-5"),
+        gr.Number(label="CA125 (U/mL) Normal range: 0-35"),
+        gr.Number(label="LDH (U/L) Normal range: 120-250"),
+    ],
+    outputs=[
+        gr.Textbox(label="预测结果"),
+        gr.Image(type="filepath", label="SHAP Waterfall Plot"),
+    ],
+    title="淋巴结转移预测",
+    description="输入变量，获取预测阳性概率及SHAP解释图",
+)
+
+demo.launch(share=True)

xgb_model.json

@@ -0,0 +1 @@
+{"learner":{"attributes":{"scikit_learn":"{\"_estimator_type\": \"classifier\"}"},"feature_names":["Tumor size(cm)","ctDNA","CT value(HU)","CEA","CYFRA21-1","CA125","LDH","Location"],"feature_types":["float","int","float","float","float","float","int","int"],"gradient_booster":{"model":{"gbtree_model_param":{"num_parallel_tree":"1","num_trees":"11"},"iteration_indptr":[0,1,2,3,4,5,6,7,8,9,10,11],"tree_info":[0,0,0,0,0,0,0,0,0,0,0],"trees":[{"base_weights":[8.967671E-3,-1.8822557E0,5.66284E-1,-2.0607235E0,1.7598717E-1,-4.7258522E-2,1.2312461E0,-4.493746E-1,5.068954E-3,-9.8007776E-2,1.0035439E-1,6.128227E-3,2.713173E-1],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0,0,0,0,0,0,0,0,0,0,0,0,0],"id":0,"left_children":[1,3,5,7,-1,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[1.13085014E2,1.3852234E1,3.386256E1,9.081024E0,0E0,1.0956368E1,5.9476204E0,0E0,0E0,0E0,0E0,0E0,0E0],"parents":[2147483647,0,0,1,1,2,2,3,3,5,5,6,6],"right_children":[2,4,6,8,-1,10,12,-1,-1,-1,-1,-1,-1],"split_conditions":[-2.5E1,1.368E1,1E0,3.8E0,1.7598717E-1,5.29E0,1.67E2,-4.493746E-1,5.068954E-3,-9.8007776E-2,1.0035439E-1,6.128227E-3,2.713173E-1],"split_indices":[2,3,7,0,0,3,6,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[1.0527902E2,2.3421625E1,8.185739E1,2.2238714E1,1.1829103E0,4.3057938E1,3.8799458E1,2.0346058E1,1.8926566E0,2.3894789E1,1.9163147E1,3.7853131E0,3.5014145E1],"tree_param":{"num_deleted":"0","num_feature":"8","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[1.4380903E-2,-1.4384952E0,4.765837E-1,-1.7333672E0,-8.425514E-2,-3.6642436E-2,1.0732287E0,-3.835398E-1,5.1298298E-2,-2.7838966E-1,2.0310004E-1,-9.758898E-2,8.5707396E-2,5.165183E-3,2.3825017E-1],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],"id":1,"left_children":[1,3,5,7,9,11,13,-1,-1,-1,-1,-1,-1,-1,-1],"loss_changes":[7.098586E1,1.0042866E1,2.4589294E1,8.158524E0,9.486352E0,9.460215E0,4.556202E0,0E0,0E0,0E0,0E0,0E0,0E0,0E0,0E0],"parents":[2147483647,0,0,1,1,2,2,3,3,4,4,5,5,6,6],"right_children":[2,4,6,8,10,12,14,-1,-1,-1,-1,-1,-1,-1,-1],"split_conditions":[-2.5E1,1.729E1,1E0,3.7E0,2.14E0,2.31E0,1.67E2,-3.835398E-1,5.1298298E-2,-2.7838966E-1,2.0310004E-1,-9.758898E-2,8.5707396E-2,5.165183E-3,2.3825017E-1],"split_indices":[2,5,7,0,4,4,6,0,0,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[1.036878E2,2.4515175E1,7.917262E1,1.9901804E1,4.613372E0,4.306622E1,3.61064E1,1.8247683E1,1.6541207E0,1.9860808E0,2.6272912E0,2.1834314E1,2.1231905E1,3.779909E0,3.2326492E1],"tree_param":{"num_deleted":"0","num_feature":"8","num_nodes":"15","size_leaf_vector":"1"}},{"base_weights":[1.341422E-2,-1.150275E0,4.274487E-1,-1.360814E0,1.8810643E-1,2.4511643E-2,9.382735E-1,-2.949831E-1,3.2591E-3,-4.771137E-2,1.6489601E-1,2.1879373E-1,2.7656138E-2],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0,0,0,0,0,0,0,0,0,0,0,0,0],"id":2,"left_children":[1,3,5,7,-1,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[4.913379E1,1.3061626E1,1.5453962E1,3.9090538E0,0E0,9.258983E0,3.9752998E0,0E0,0E0,0E0,0E0,0E0,0E0],"parents":[2147483647,0,0,1,1,2,2,3,3,5,5,6,6],"right_children":[2,4,6,8,-1,10,12,-1,-1,-1,-1,-1,-1],"split_conditions":[-9E0,1.368E1,1E0,3.8E0,1.8810643E-1,2.41E2,4.8E0,-2.949831E-1,3.2591E-3,-4.771137E-2,1.6489601E-1,2.1879373E-1,2.7656138E-2],"split_indices":[2,3,7,0,0,6,0,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[9.995756E1,2.5764736E1,7.4192825E1,2.372485E1,2.0398874E0,4.206224E1,3.213058E1,2.1843931E1,1.8809174E0,3.218127E1,9.880973E0,2.658845E1,5.542134E0],"tree_param":{"num_deleted":"0","num_feature":"8","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[1.565923E-2,-1.1323609E0,3.225917E-1,-2.922413E-1,2.367571E-1,-7.1443826E-2,8.5827243E-1,-2.0796521E-1,2.4229436E-1,7.531268E-3,-3.3670944E-1,-2.0827452E-2,1.9622223E-1],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0,0,0,0,0,0,0,0,0,0,0,0,0],"id":3,"left_children":[1,3,5,-1,7,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[3.4426632E1,9.626905E0,1.6325064E1,0E0,7.2981653E0,8.0339985E0,3.9718952E0,0E0,0E0,0E0,0E0,0E0,0E0],"parents":[2147483647,0,0,1,1,2,2,4,4,5,5,6,6],"right_children":[2,4,6,-1,8,10,12,-1,-1,-1,-1,-1,-1],"split_conditions":[-4.8E1,1.729E1,1E0,-2.922413E-1,2.14E0,4.5E0,1.67E2,-2.0796521E-1,2.4229436E-1,7.531268E-3,-3.3670944E-1,-2.0827452E-2,1.9622223E-1],"split_indices":[2,5,7,0,4,0,6,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[9.566504E1,1.9613888E1,7.6051155E1,1.576102E1,3.8528686E0,4.431823E1,3.1732925E1,1.6392039E0,2.2136645E0,4.242381E1,1.8944181E0,3.6090665E0,2.812386E1],"tree_param":{"num_deleted":"0","num_feature":"8","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[9.536208E-3,-9.0577817E-1,3.1588754E-1,-1.0720685E0,1.5810771E-1,9.052059E-2,9.474351E-1,-2.6830038E-1,-1.0381501E-1,-1.1541616E-1,8.052449E-2,2.148412E-1,1.3686098E-2],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0,0,0,0,0,0,0,0,0,0,0,0,0],"id":4,"left_children":[1,3,5,7,-1,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[2.6107857E1,7.434286E0,9.839722E0,2.5354118E0,0E0,1.1171433E1,1.9584293E0,0E0,0E0,0E0,0E0,0E0,0E0],"parents":[2147483647,0,0,1,1,2,2,3,3,5,5,6,6],"right_children":[2,4,6,8,-1,10,12,-1,-1,-1,-1,-1,-1],"split_conditions":[-9E0,1.368E1,1E0,1.313E1,1.5810771E-1,2.4E0,6.9E0,-2.6830038E-1,-1.0381501E-1,-1.1541616E-1,8.052449E-2,2.148412E-1,1.3686098E-2],"split_indices":[2,3,1,5,0,0,3,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[9.1086365E1,2.2350636E1,6.873573E1,2.0690355E1,1.6602787E0,5.1500996E1,1.7234735E1,1.2953279E1,7.7370777E0,1.6136087E1,3.536491E1,1.4868339E1,2.3663962E0],"tree_param":{"num_deleted":"0","num_feature":"8","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[8.062089E-3,-8.025019E-1,2.6646295E-1,-9.5507777E-1,1.4767481E-1,-2.575125E-3,6.916458E-1,-2.544659E-1,-8.2638934E-2,1.3032246E-1,-5.841434E-2,1.747388E-1,-2.0444652E-2],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0,0,0,0,0,0,0,0,0,0,0,0,0],"id":5,"left_children":[1,3,5,7,-1,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[1.870246E1,5.780381E0,7.744269E0,2.9260159E0,0E0,8.229815E0,3.880746E0,0E0,0E0,0E0,0E0,0E0,0E0],"parents":[2147483647,0,0,1,1,2,2,3,3,5,5,6,6],"right_children":[2,4,6,8,-1,10,12,-1,-1,-1,-1,-1,-1],"split_conditions":[-9E0,1.368E1,1E0,2.12E2,1.4767481E-1,2.3E1,4.8E0,-2.544659E-1,-8.2638934E-2,1.3032246E-1,-5.841434E-2,1.747388E-1,-2.0444652E-2],"split_indices":[2,3,7,6,0,2,0,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[8.727206E1,2.0587275E1,6.6684784E1,1.9078333E1,1.5089432E0,4.1450542E1,2.523424E1,1.1183086E1,7.8952456E0,1.2332173E1,2.911837E1,2.0445238E1,4.789003E0],"tree_param":{"num_deleted":"0","num_feature":"8","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[7.790033E-3,-3.1876814E-1,5.058186E-1,-6.372522E-1,4.8892692E-1,-4.3318456E-1,6.9131124E-1,-1.9049023E-1,1.2363386E-1,-7.5073336E-4,1.7321022E-1,1.28937E-1,-2.890769E-1,1.6864543E-1,-9.463193E-2],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],"id":6,"left_children":[1,3,5,7,9,11,13,-1,-1,-1,-1,-1,-1,-1,-1],"loss_changes":[1.3917335E1,1.3610518E1,6.201229E0,1.5432809E1,2.8249028E0,7.700577E0,5.472414E0,0E0,0E0,0E0,0E0,0E0,0E0,0E0,0E0],"parents":[2147483647,0,0,1,1,2,2,3,3,4,4,5,5,6,6],"right_children":[2,4,6,8,10,12,14,-1,-1,-1,-1,-1,-1,-1,-1],"split_conditions":[1.603E1,1E0,1.3E1,2.45E2,4.1E1,-1.81E2,4.8E0,-1.9049023E-1,1.2363386E-1,-7.5073336E-4,1.7321022E-1,1.28937E-1,-2.890769E-1,1.6864543E-1,-9.463193E-2],"split_indices":[5,7,2,6,2,2,0,0,0,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[8.358247E1,5.0698997E1,3.288348E1,3.6512638E1,1.41863575E1,5.204068E0,2.7679413E1,2.9378023E1,7.1346164E0,6.5800843E0,7.606273E0,2.6961172E0,2.5079508E0,2.4729141E1,2.9502704E0],"tree_param":{"num_deleted":"0","num_feature":"8","num_nodes":"15","size_leaf_vector":"1"}},{"base_weights":[8.686729E-3,-8.0134904E-1,1.8409878E-1,-2.2890607E-1,3.2771215E-1,-2.4825653E-1,4.4310448E-1,-1.5237571E-1,2.1563919E-1,-1.0781486E-1,1.5870024E-1,1.3170426E-1,-7.334577E-2],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0,0,0,0,0,0,0,0,0,0,0,0,0],"id":7,"left_children":[1,3,5,-1,7,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[1.1664398E1,6.037796E0,7.661971E0,0E0,4.2428465E0,8.276171E0,7.6614895E0,0E0,0E0,0E0,0E0,0E0,0E0],"parents":[2147483647,0,0,1,1,2,2,4,4,5,5,6,6],"right_children":[2,4,6,-1,8,10,12,-1,-1,-1,-1,-1,-1],"split_conditions":[-4.8E1,1.729E1,2.5E0,-2.2890607E-1,2.14E0,2.45E2,4.5E0,-1.5237571E-1,2.1563919E-1,-1.0781486E-1,1.5870024E-1,1.3170426E-1,-7.334577E-2],"split_indices":[2,5,0,0,4,6,0,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[8.005333E1,1.3614502E1,6.643883E1,1.043044E1,3.1840622E0,2.4905632E1,4.153319E1,1.2924435E0,1.8916187E0,1.9847519E1,5.058113E0,3.2954178E1,8.579016E0],"tree_param":{"num_deleted":"0","num_feature":"8","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[6.6590915E-3,-2.3169076E-1,5.2605885E-1,-5.168103E-1,2.666196E-1,2.5208715E-1,2.0820741E-1,-1.5242445E-1,9.538922E-2,-2.1587794E-1,8.370162E-2,1.03100516E-1,-1.9051825E-1],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0,0,0,0,0,0,0,0,0,0,0,0,0],"id":8,"left_children":[1,3,5,7,9,11,-1,-1,-1,-1,-1,-1,-1],"loss_changes":[9.7657585E0,7.8228655E0,3.2378616E0,8.833705E0,4.6111827E0,6.0485873E0,0E0,0E0,0E0,0E0,0E0,0E0,0E0],"parents":[2147483647,0,0,1,1,2,2,3,3,4,4,5,5],"right_children":[2,4,6,8,10,12,-1,-1,-1,-1,-1,-1,-1],"split_conditions":[1E0,1.58E1,1E0,2.45E2,1.54E0,4.7E0,2.0820741E-1,-1.5242445E-1,9.538922E-2,-2.1587794E-1,8.370162E-2,1.03100516E-1,-1.9051825E-1],"split_indices":[7,5,1,6,4,0,0,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[7.689955E1,5.3090515E1,2.3809036E1,3.374527E1,1.9345245E1,1.6513342E1,7.2956934E0,2.725129E1,6.4939795E0,1.3424275E0,1.8002817E1,1.4019841E1,2.4935012E0],"tree_param":{"num_deleted":"0","num_feature":"8","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[6.409361E-3,-7.32994E-1,1.4918748E-1,-2.1394475E-1,2.609475E-1,-3.2583093E-3,6.7859256E-1,-1.4170817E-1,1.897102E-1,-4.285203E-2,1.4103849E-1,1.631705E-1,-5.2152105E-2],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0,0,0,0,0,0,0,0,0,0,0,0,0],"id":9,"left_children":[1,3,5,-1,7,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[8.011798E0,4.390494E0,5.1327057E0,0E0,3.2801943E0,7.6780624E0,2.0459027E0,0E0,0E0,0E0,0E0,0E0,0E0],"parents":[2147483647,0,0,1,1,2,2,4,4,5,5,6,6],"right_children":[2,4,6,-1,8,10,12,-1,-1,-1,-1,-1,-1],"split_conditions":[-4.8E1,1.729E1,1E0,-2.1394475E-1,2.14E0,1.067E1,2.696E1,-1.4170817E-1,1.897102E-1,-4.285203E-2,1.4103849E-1,1.631705E-1,-5.2152105E-2],"split_indices":[2,5,1,0,4,3,5,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[7.385458E1,1.1284076E1,6.257051E1,8.371661E0,2.9124136E0,4.9352863E1,1.3217646E1,1.1958592E0,1.7165544E0,3.8564537E1,1.0788326E1,1.1647209E1,1.5704367E0],"tree_param":{"num_deleted":"0","num_feature":"8","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[6.1202794E-3,-1.913991E-1,4.522438E-1,-7.672764E-1,-1.3416311E-2,2.213648E-1,1.924303E-1,-2.0796132E-1,-1.932215E-2,2.2782128E-1,-3.827154E-2,1.5834013E-1,-5.077346E-2],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0,0,0,0,0,0,0,0,0,0,0,0,0],"id":10,"left_children":[1,3,5,7,9,11,-1,-1,-1,-1,-1,-1,-1],"loss_changes":[6.4493103E0,5.191497E0,2.4307022E0,2.0863342E0,8.377503E0,4.742589E0,0E0,0E0,0E0,0E0,0E0,0E0,0E0],"parents":[2147483647,0,0,1,1,2,2,3,3,4,4,5,5],"right_children":[2,4,6,8,10,12,-1,-1,-1,-1,-1,-1,-1],"split_conditions":[1E0,-9E0,1E0,1.645E1,5E0,2.05E0,1.924303E-1,-2.0796132E-1,-1.932215E-2,2.2782128E-1,-3.827154E-2,1.5834013E-1,-5.077346E-2],"split_indices":[7,2,1,5,2,4,0,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[7.120713E1,4.9751038E1,2.1456097E1,1.0964282E1,3.8786755E1,1.5756088E1,5.700008E0,7.404853E0,3.5594294E0,4.465061E0,3.4321693E1,6.8587894E0,8.897299E0],"tree_param":{"num_deleted":"0","num_feature":"8","num_nodes":"13","size_leaf_vector":"1"}}]},"name":"gbtree"},"learner_model_param":{"base_score":"6.158358E-1","boost_from_average":"1","num_class":"0","num_feature":"8","num_target":"1"},"objective":{"name":"binary:logistic","reg_loss_param":{"scale_pos_weight":"5"}}},"version":[2,1,4]}