1/xgboost.json

@@ -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]}

config.pbtxt

@@ -1,45 +1,20 @@
-backend: "fil",
+# Triton backend to use
+name: "XGBoost"
+backend: "fil"
 max_batch_size: 0
-default_model_filename: "xgboost.json"
 
-input [                                 
+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"
+    kind: KIND_AUTO
-    value: { string_value: "xgboost_json" }
+    count: 1
-  },
-  {
-    key: "predict_proba"
-    value: { string_value: "false" }
-  },
-  {
-    key: "is_classifier"
-    value: { string_value: "true" }
-  },
-  {
-    key: "threshold"
-    value: { string_value: "0.5" }
   }
 ]