groupuser/XGBoost
1
0
Fork 0
Code Issues Pull Requests Actions Packages Projects Releases Wiki Activity

Update config.pbtxt

Browse Source
This commit is contained in:
groupuser 2025-09-29 06:50:23 +00:00
parent 3a522b6d89
commit cc359c2be6
1 changed files with 9 additions and 12 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

21
config.pbtxt
View File

@ -1,12 +1,13 @@
backend: "fil", name: "xgboost-triton-fil"
backend: "fil"
max_batch_size: 0 max_batch_size: 0
default_model_filename: "xgboost.json" default_model_filename: "xgboost.json"
input [ input [
{ {
name: "input__0" name: "input__0"
data_type: TYPE_FP32 data_type: TYPE_FP32
dims: [ 3 ] dims: [ 3 ] # feature 개수
} }
] ]
@ -18,14 +19,11 @@ output [
} }
] ]
parameters {
key: "output_class"
value: {
string_value: "true"
}
}
parameters [ parameters [
{
key: "output_class"
value: { string_value: "true" }
},
{ {
key: "model_type" key: "model_type"
value: { string_value: "xgboost_json" } value: { string_value: "xgboost_json" }
@ -43,4 +41,3 @@ parameters [
value: { string_value: "0.5" } value: { string_value: "0.5" }
} }
] ]