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

Compare commits

base: groupuser:main
Branches Tags
groupuser:main
groupuser:refs/deployment/triton
..
compare: groupuser:refs/deployment/triton
Branches Tags
groupuser:main
groupuser:refs/deployment/triton

No commits in common. "main" and "refs/deployment/triton" have entirely different histories.
main ... refs/deplo

2 changed files with 9 additions and 35 deletions
Show Stats Expand all files Collapse all files

1
1/xgboost.json
View File

@ -1 +0,0 @@
{"learner":{"attributes":{"scikit_learn":"{\"_estimator_type\": \"classifier\"}"},"feature_names":["age","income","city"],"feature_types":["int","int","c"],"gradient_booster":{"model":{"gbtree_model_param":{"num_parallel_tree":"1","num_trees":"5"},"iteration_indptr":[0,1,2,3,4,5],"tree_info":[0,0,0,0,0],"trees":[{"base_weights":[-5.4186042E-8],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0],"id":0,"left_children":[-1],"loss_changes":[0E0],"parents":[2147483647],"right_children":[-1],"split_conditions":[-1.6255813E-8],"split_indices":[0],"split_type":[0],"sum_hessian":[1.1999999E0],"tree_param":{"num_deleted":"0","num_feature":"3","num_nodes":"1","size_leaf_vector":"1"}},{"base_weights":[-5.4186042E-8],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0],"id":1,"left_children":[-1],"loss_changes":[0E0],"parents":[2147483647],"right_children":[-1],"split_conditions":[-1.6255813E-8],"split_indices":[0],"split_type":[0],"sum_hessian":[1.1999999E0],"tree_param":{"num_deleted":"0","num_feature":"3","num_nodes":"1","size_leaf_vector":"1"}},{"base_weights":[8.127906E-8],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0],"id":2,"left_children":[-1],"loss_changes":[0E0],"parents":[2147483647],"right_children":[-1],"split_conditions":[2.438372E-8],"split_indices":[0],"split_type":[0],"sum_hessian":[1.2E0],"tree_param":{"num_deleted":"0","num_feature":"3","num_nodes":"1","size_leaf_vector":"1"}},{"base_weights":[-5.4186042E-8],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0],"id":3,"left_children":[-1],"loss_changes":[0E0],"parents":[2147483647],"right_children":[-1],"split_conditions":[-1.6255813E-8],"split_indices":[0],"split_type":[0],"sum_hessian":[1.1999999E0],"tree_param":{"num_deleted":"0","num_feature":"3","num_nodes":"1","size_leaf_vector":"1"}},{"base_weights":[8.127906E-8],"categories":[],"categories_nodes":[],"categories_segments":[],"categories_sizes":[],"default_left":[0],"id":4,"left_children":[-1],"loss_changes":[0E0],"parents":[2147483647],"right_children":[-1],"split_conditions":[2.438372E-8],"split_indices":[0],"split_type":[0],"sum_hessian":[1.2E0],"tree_param":{"num_deleted":"0","num_feature":"3","num_nodes":"1","size_leaf_vector":"1"}}]},"name":"gbtree"},"learner_model_param":{"base_score":"6E-1","boost_from_average":"1","num_class":"0","num_feature":"3","num_target":"1"},"objective":{"name":"binary:logistic","reg_loss_param":{"scale_pos_weight":"1"}}},"version":[3,0,5]}

41
config.pbtxt
View File

@ -1,45 +1,20 @@
backend: "fil",
# Triton backend to use
name: "XGBoost"
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 [
instance_group [
{
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" }
kind: KIND_AUTO
count: 1
}
]