XGBoost/config.pbtxt
2025-09-29 06:27:56 +00:00

47 lines
647 B
Plaintext

backend: "fil",
max_batch_size: 0
default_model_filename: "xgboost.json"
input [
{
name: "input__0"
data_type: TYPE_FP32
dims: [ 3 ]
}
]
output [
{
name: "output__0"
data_type: TYPE_FP32
dims: [ 1 ]
}
]
parameters {
key: "output_class"
value: {
string_value: "true"
}
}
parameters [
{
key: "model_type"
value: { string_value: "xgboost_json" }
},
{
key: "predict_proba"
value: { string_value: "false" }
},
{
key: "is_classifier"
value: { string_value: "true" }
},
{
key: "threshold"
value: { string_value: "0.5" }
}
]