Spaces:
Sleeping
Sleeping
Upload 2 files
Browse files15-feature model
- app.py +133 -0
- xgb_model.json +1 -0
app.py
ADDED
@@ -0,0 +1,133 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
import numpy as np
|
2 |
+
import pandas as pd
|
3 |
+
import xgboost as xgb
|
4 |
+
import shap
|
5 |
+
import gradio as gr
|
6 |
+
import matplotlib.pyplot as plt
|
7 |
+
import matplotlib
|
8 |
+
|
9 |
+
matplotlib.use("Agg") # 防止服务器无图形界面时报错
|
10 |
+
|
11 |
+
# 指定输入特征顺序
|
12 |
+
feature_names = [
|
13 |
+
"CT value(HU)",
|
14 |
+
"Tumor size(cm)",
|
15 |
+
"ctDNA",
|
16 |
+
"CEA",
|
17 |
+
"CYFRA21-1",
|
18 |
+
"AAPR",
|
19 |
+
"Location",
|
20 |
+
"CYFRA21-1",
|
21 |
+
"CA125",
|
22 |
+
"LDH",
|
23 |
+
"ANC",
|
24 |
+
"ALT",
|
25 |
+
"GGT",
|
26 |
+
"CREA",
|
27 |
+
"UREA",
|
28 |
+
"Pleural indentation",
|
29 |
+
]
|
30 |
+
|
31 |
+
# 加载模型(确保是用 sklearn API 训练并保存的)
|
32 |
+
model = xgb.XGBClassifier()
|
33 |
+
model.load_model("./xgb_model.json")
|
34 |
+
model.get_booster().feature_names = feature_names
|
35 |
+
|
36 |
+
# 初始化 SHAP 解释器
|
37 |
+
explainer = shap.Explainer(model)
|
38 |
+
|
39 |
+
|
40 |
+
# 预测函数
|
41 |
+
def predict_probability(
|
42 |
+
CT_value,
|
43 |
+
Tumor_size,
|
44 |
+
ctDNA,
|
45 |
+
CEA,
|
46 |
+
Location,
|
47 |
+
CYFRA21_1,
|
48 |
+
AAPR,
|
49 |
+
CA125,
|
50 |
+
LDH,
|
51 |
+
ANC,
|
52 |
+
ALT,
|
53 |
+
GGT,
|
54 |
+
CREA,
|
55 |
+
UREA,
|
56 |
+
Pleural_indentation,
|
57 |
+
):
|
58 |
+
input_data = pd.DataFrame(
|
59 |
+
[
|
60 |
+
[
|
61 |
+
CT_value,
|
62 |
+
Tumor_size,
|
63 |
+
ctDNA,
|
64 |
+
CEA,
|
65 |
+
Location,
|
66 |
+
CYFRA21_1,
|
67 |
+
AAPR,
|
68 |
+
CA125,
|
69 |
+
LDH,
|
70 |
+
ANC,
|
71 |
+
ALT,
|
72 |
+
GGT,
|
73 |
+
CREA,
|
74 |
+
UREA,
|
75 |
+
Pleural_indentation,
|
76 |
+
]
|
77 |
+
],
|
78 |
+
columns=feature_names,
|
79 |
+
)
|
80 |
+
|
81 |
+
# 将 Location 和 ctDNA 转换为数值型
|
82 |
+
input_data["Location"] = input_data["Location"].map({"Central": 1, "Peripheral": 0})
|
83 |
+
input_data["ctDNA"] = input_data["ctDNA"].map({"Positive": 1, "Negative": 0})
|
84 |
+
input_data["Pleural_indentation"] = input_data["Pleural_indentation"].map({"Positive": 1, "Negative": 0})
|
85 |
+
# 预测
|
86 |
+
try:
|
87 |
+
prob = model.predict_proba(input_data)[0][1]
|
88 |
+
except Exception as e:
|
89 |
+
return f"预测出错: {e}", None
|
90 |
+
|
91 |
+
# 计算 SHAP 值
|
92 |
+
try:
|
93 |
+
shap_values = explainer(input_data)
|
94 |
+
|
95 |
+
# 绘图
|
96 |
+
shap.plots.waterfall(shap_values[0], show=False)
|
97 |
+
plt.title("SHAP Waterfall Plot")
|
98 |
+
plt.savefig("shap_plot.png", bbox_inches="tight", dpi=300)
|
99 |
+
plt.close()
|
100 |
+
except Exception as e:
|
101 |
+
return f"SHAP 图生成失败: {e}", None
|
102 |
+
|
103 |
+
return f"阳性概率: {prob:.2%}", "shap_plot.png"
|
104 |
+
|
105 |
+
|
106 |
+
demo = gr.Interface(
|
107 |
+
fn=predict_probability,
|
108 |
+
inputs=[
|
109 |
+
gr.Number(label="CT value(HU)"),
|
110 |
+
gr.Number(label="Tumor size(cm)"),
|
111 |
+
gr.Dropdown(choices=["Positive", "Negative"], label="ctDNA"),
|
112 |
+
gr.Number(label="CEA (ng/mL) Normal range: 0-5"),
|
113 |
+
gr.Dropdown(choices=["Central", "Peripheral"], label="Location"), # 修改为 Dropdown 类型
|
114 |
+
gr.Number(label="CYFRA21-1 (ng/mL) Normal range: 0-5"),
|
115 |
+
gr.Number(label="AAPR (ng/mL) Normal range: 0-5"),
|
116 |
+
gr.Number(label="CA125 (U/mL) Normal range: 0-35"),
|
117 |
+
gr.Number(label="LDH (U/L) Normal range: 120-250"),
|
118 |
+
gr.Number(label="ANC (10^9/L) Normal range: 1.8-6.3"),
|
119 |
+
gr.Number(label="ALT (U/L) Normal range: 7-40"),
|
120 |
+
gr.Number(label="GGT (U/L) Normal range: 7-45"),
|
121 |
+
gr.Number(label="CREA (μmol/L) Normal range: 41-81"),
|
122 |
+
gr.Number(label="UREA (mmol/L) Normal range: 2.6-8.8"),
|
123 |
+
gr.Dropdown(choices=["Positive", "Negative"], label="Pleural indentation"),
|
124 |
+
],
|
125 |
+
outputs=[
|
126 |
+
gr.Textbox(label="Results of prediction"),
|
127 |
+
gr.Image(type="filepath", label="SHAP Waterfall Plot"),
|
128 |
+
],
|
129 |
+
title="Prediction of Lymph Node Metastasis",
|
130 |
+
description="Variables were entered to obtain the predicted positive probability and SHAP interpretation map",
|
131 |
+
)
|
132 |
+
|
133 |
+
demo.launch(share=True)
|
xgb_model.json
ADDED
@@ -0,0 +1 @@
|
|
|
|
|
1 |
+
{"learner":{"attributes":{"scikit_learn":"{\"_estimator_type\": \"classifier\"}"},"feature_names":["Tumor size(cm)","ctDNA","CT value(HU)","CEA","CYFRA21-1","CA125","AAPR","LDH","Location","ANC","ALT","GGT","CREA","UREA","Pleural indentation"],"feature_types":["float","int","float","float","float","float","float","int","int","float","float","int","int","float","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":[2.3006016E-2,-2.0278656E0,5.6512046E-1,-2.240275E0,1.7582415E-1,-3.3316787E-2,1.1811181E0,1.0972337E-1,-4.950435E-1,7.650243E-2,-1.4600219E-1,8.460886E-3,2.5873896E-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.2792056E2,1.6824135E1,3.357359E1,1.6232239E1,0E0,1.3928564E1,5.695713E0,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":[6.5514483E0,1.368E1,1E0,3.4285715E-1,1.7582415E-1,5.4761904E-1,1.67E2,1.0972337E-1,-4.950435E-1,7.650243E-2,-1.4600219E-1,8.460886E-3,2.5873896E-1],"split_indices":[2,3,8,6,0,6,7,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[1.1302569E2,2.305724E1,8.9968445E1,2.1700932E1,1.3563082E0,4.611448E1,4.3853966E1,1.5823596E0,2.0118572E1,2.9160627E1,1.6953854E1,4.068925E0,3.978504E1],"tree_param":{"num_deleted":"0","num_feature":"15","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[2.8676169E-2,-1.6590545E0,4.5139226E-1,-4.100123E-1,2.2471831E-3,-9.185154E-2,1.0548776E0,-2.7387452E-1,2.1015842E-1,-1.13935605E-1,1.021917E-1,-7.243899E-2,2.2698243E-1],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0,0,0,0,0,0,0,0,0,0,0,0,0],"id":1,"left_children":[1,3,5,-1,7,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[8.08473E1,1.47511635E1,2.9789488E1,0E0,9.09598E0,1.4285184E1,5.2863426E0,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":[6.5512395E0,1.729E1,1E0,-4.100123E-1,2.14E0,5.29E0,1.9E1,-2.7387452E-1,2.1015842E-1,-1.13935605E-1,1.021917E-1,-7.243899E-2,2.2698243E-1],"split_indices":[2,5,8,0,4,3,0,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[1.1127213E2,2.1701653E1,8.957048E1,1.7377617E1,4.324035E0,4.758413E1,4.198635E1,1.7408392E0,2.5831962E0,2.6574451E1,2.1009676E1,2.056096E0,3.9930256E1],"tree_param":{"num_deleted":"0","num_feature":"15","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[2.5763627E-2,-1.3706039E0,3.836164E-1,-3.3570835E-1,-5.8932663E-3,-3.9761594E-1,7.385338E-1,-2.3879002E-1,1.9274041E-1,-2.3676725E-1,6.933934E-2,1.610849E-1,-2.815539E-1],"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,-1,7,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[5.462216E1,9.362865E0,2.4404198E1,0E0,7.036478E0,1.6686935E1,9.768318E0,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":[6.5512395E0,1.729E1,2.1E1,-3.3570835E-1,2.14E0,7.1E1,7.818182E-1,-2.3879002E-1,1.9274041E-1,-2.3676725E-1,6.933934E-2,1.610849E-1,-2.815539E-1],"split_indices":[2,5,0,0,4,12,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.07259E2,2.1303013E1,8.595599E1,1.7193462E1,4.10955E0,2.6813879E1,5.9142113E1,1.7427489E0,2.3668013E0,1.2752673E1,1.4061205E1,5.7960484E1,1.1816303E0],"tree_param":{"num_deleted":"0","num_feature":"15","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[1.7642641E-2,-1.047166E0,3.7136388E-1,-1.2496438E0,1.735398E-1,6.501605E-3,8.479426E-1,8.011E-2,-2.7315268E-1,-5.6069452E-2,1.6732594E-1,-4.8001982E-2,1.9673716E-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,7,-1,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[3.9225975E1,1.1154837E1,1.3581785E1,5.126732E0,0E0,1.1013006E1,5.238785E0,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":[6.5515933E0,1.368E1,1E0,3.4285715E-1,1.735398E-1,2.41E2,1.67E2,8.011E-2,-2.7315268E-1,-5.6069452E-2,1.6732594E-1,-4.8001982E-2,1.9673716E-1],"split_indices":[2,3,8,6,0,7,7,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[1.02113144E2,2.4973843E1,7.7139305E1,2.2899815E1,2.0740275E0,4.426458E1,3.2874725E1,1.3444316E0,2.1555384E1,3.3389275E1,1.0875303E1,3.5615487E0,2.9313175E1],"tree_param":{"num_deleted":"0","num_feature":"15","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[1.9363372E-2,-1.0799745E0,2.823674E-1,-2.6742062E-1,2.902474E-2,-8.4255174E-2,7.96959E-1,-1.87097E-1,1.6961001E-1,6.530784E-3,-3.597072E-1,-1.3012998E-1,1.7888936E-1],"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,-1,7,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[2.8706497E1,5.505533E0,1.5171323E1,0E0,4.4169297E0,9.609701E0,5.2960777E0,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":[6.5512395E0,1.729E1,1E0,-2.6742062E-1,2.14E0,6.4E0,1E0,-1.87097E-1,1.6961001E-1,6.530784E-3,-3.597072E-1,-1.3012998E-1,1.7888936E-1],"split_indices":[2,5,8,0,4,9,14,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[9.723096E1,1.8171036E1,7.905992E1,1.4583966E1,3.5870697E0,4.6655975E1,3.240395E1,1.5819377E0,2.005132E0,4.459563E1,2.060344E0,1.6865147E0,3.0717434E1],"tree_param":{"num_deleted":"0","num_feature":"15","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[1.104228E-2,-9.9194384E-1,2.3473886E-1,-2.4673402E-1,2.2460004E-2,-3.3446035E-1,5.385832E-1,-1.7156652E-1,1.5832071E-1,-2.4371484E-1,3.9480843E-2,1.21229105E-1,-2.381664E-1],"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,-1,7,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[2.11589E1,4.2451973E0,1.3508968E1,0E0,3.5850337E0,1.3143138E1,6.6375313E0,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":[6.5512395E0,1.729E1,2.1E1,-2.4673402E-1,2.14E0,2.08E0,7.818182E-1,-1.7156652E-1,1.5832071E-1,-2.4371484E-1,3.9480843E-2,1.21229105E-1,-2.381664E-1],"split_indices":[2,5,0,0,4,4,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.226808E1,1.619963E1,7.606845E1,1.29076605E1,3.2919703E0,2.6438948E1,4.96295E1,1.4812994E0,1.8106709E0,9.445928E0,1.6993021E1,4.8388634E1,1.2408679E0],"tree_param":{"num_deleted":"0","num_feature":"15","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[3.136371E-3,-7.8794247E-1,2.3140468E-1,-9.495455E-1,1.4195798E-1,5.2107433E-3,8.724738E-1,-2.5078282E-1,-3.553324E-2,-7.9195336E-2,1.03416264E-1,2.0727001E-1,-4.8454363E-2],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0,0,0,0,0,0,0,0,0,0,0,0,0],"id":6,"left_children":[1,3,5,7,-1,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[1.615921E1,5.5244226E0,1.0064117E1,4.127308E0,0E0,1.0951508E1,3.5583E0,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":[6.5515933E0,1.368E1,1E0,2.35E1,1.4195798E-1,1.58E1,4.7E1,-2.5078282E-1,-3.553324E-2,-7.9195336E-2,1.03416264E-1,2.0727001E-1,-4.8454363E-2],"split_indices":[2,3,1,0,0,5,11,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[8.74762E1,1.904E1,6.84362E1,1.7516056E1,1.5239449E0,5.1332306E1,1.7103891E1,1.2114539E1,5.4015174E0,2.8904398E1,2.2427908E1,1.49730625E1,2.1308296E0],"tree_param":{"num_deleted":"0","num_feature":"15","num_nodes":"13","size_leaf_vector":"1"}},{"base_weights":[4.0757842E-3,-2.676496E-1,5.656663E-1,-9.4851035E-1,-2.585036E-2,-8.1035095E-1,7.120517E-1,6.484654E-2,-2.1464102E-1,1.1158142E-1,-8.226693E-2,-3.0349585E-1,5.6023072E-2,-2.710406E-2,1.7500818E-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":7,"left_children":[1,3,5,7,9,11,13,-1,-1,-1,-1,-1,-1,-1,-1],"loss_changes":[1.3089839E1,9.494322E0,6.2106657E0,2.74829E0,1.0062045E1,2.8594306E0,3.6843119E0,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":[1E0,6.5515933E0,1E0,3.4285715E-1,6.5519023E0,1.8E1,1.69E2,6.484654E-2,-2.1464102E-1,1.1158142E-1,-8.226693E-2,-3.0349585E-1,5.6023072E-2,-2.710406E-2,1.7500818E-1],"split_indices":[8,2,14,6,2,10,7,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.3787506E1,5.681904E1,2.6968464E1,1.4124477E1,4.269456E1,2.1570003E0,2.4811464E1,1.1175536E0,1.3006924E1,1.6802473E1,2.5892088E1,1.0709442E0,1.0860561E0,4.0289197E0,2.0782545E1],"tree_param":{"num_deleted":"0","num_feature":"15","num_nodes":"15","size_leaf_vector":"1"}},{"base_weights":[2.2205648E-3,-4.0335768E-1,2.8910962E-1,-9.167492E-1,1.9543068E-1,4.2858177E-1,-6.768037E-1,-2.6502225E-1,1.9090353E-1,-2.0212176E-1,1.5940844E-1,1.20842956E-1,-1.2508722E-1,-3.236349E-1,1.0890807E-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":8,"left_children":[1,3,5,7,9,11,13,-1,-1,-1,-1,-1,-1,-1,-1],"loss_changes":[9.565351E0,1.06499195E1,6.776E0,1.5412258E1,1.2811514E1,8.279234E0,8.311184E0,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.15E1,1.343E1,4.25E1,2.55E2,6.1E0,8.7E1,5.4E0,-2.6502225E-1,1.9090353E-1,-2.0212176E-1,1.5940844E-1,1.20842956E-1,-1.2508722E-1,-3.236349E-1,1.0890807E-1],"split_indices":[0,5,0,7,13,12,9,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.020543E1,3.30609E1,4.714453E1,1.7513823E1,1.5547077E1,4.1682167E1,5.4623632E0,1.461606E1,2.8977628E0,4.9480257E0,1.0599051E1,3.6094433E1,5.5877337E0,2.9012585E0,2.5611048E0],"tree_param":{"num_deleted":"0","num_feature":"15","num_nodes":"15","size_leaf_vector":"1"}},{"base_weights":[-2.2736244E-4,-2.2469255E-1,4.86186E-1,7.99039E-2,-6.428751E-1,-7.6145273E-1,6.307633E-1,-7.177226E-2,1.2483119E-1,-3.1217816E-1,-1.782055E-2,-2.2234432E-1,-9.53997E-3,1.4588222E-1,-1.0581678E-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":9,"left_children":[1,3,5,7,9,11,13,-1,-1,-1,-1,-1,-1,-1,-1],"loss_changes":[8.657808E0,6.9624934E0,4.9967623E0,7.9327483E0,1.1541205E1,7.1029794E-1,3.0158186E0,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":[1E0,5.4761904E-1,1E0,4.659091E-1,3.9E0,6.3E0,6.5753424E-1,-7.177226E-2,1.2483119E-1,-3.1217816E-1,-1.782055E-2,-2.2234432E-1,-9.53997E-3,1.4588222E-1,-1.0581678E-1],"split_indices":[8,6,14,6,9,13,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":[7.729586E1,5.3257282E1,2.4038578E1,3.1281406E1,2.1975876E1,2.0532422E0,2.1985334E1,1.7507689E1,1.3773718E1,7.584321E0,1.4391554E1,1.0033056E0,1.0499368E0,2.0604027E1,1.3813077E0],"tree_param":{"num_deleted":"0","num_feature":"15","num_nodes":"15","size_leaf_vector":"1"}},{"base_weights":[-5.1551964E-3,-7.6135164E-1,1.2342054E-1,-2.0581396E-1,7.668243E-2,-2.7475031E-2,6.511811E-1,-1.3644044E-1,1.3990153E-1,2.0957598E-2,-2.4667738E-1,1.6123825E-1,-6.7095935E-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,-1,7,9,11,-1,-1,-1,-1,-1,-1],"loss_changes":[7.423451E0,2.5680094E0,5.217353E0,0E0,2.1866732E0,8.380439E0,2.496891E0,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":[6.5512395E0,1.729E1,1E0,-2.0581396E-1,2.14E0,5.2E0,6.9E0,-1.3644044E-1,1.3990153E-1,2.0957598E-2,-2.4667738E-1,1.6123825E-1,-6.7095935E-2],"split_indices":[2,5,1,0,4,9,3,0,0,0,0,0,0],"split_type":[0,0,0,0,0,0,0,0,0,0,0,0,0],"sum_hessian":[7.438411E1,1.0094367E1,6.428974E1,7.4776134E0,2.616753E0,5.073243E1,1.3557309E1,1.136565E0,1.4801883E0,4.6540974E1,4.191456E0,1.1874642E1,1.6826661E0],"tree_param":{"num_deleted":"0","num_feature":"15","num_nodes":"13","size_leaf_vector":"1"}}]},"name":"gbtree"},"learner_model_param":{"base_score":"6.5475345E-1","boost_from_average":"1","num_class":"0","num_feature":"15","num_target":"1"},"objective":{"name":"binary:logistic","reg_loss_param":{"scale_pos_weight":"6"}}},"version":[2,1,4]}
|