1/xgboost.json

@@ -0,0 +1 @@
+{"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,20 +1,45 @@
-# Triton backend to use
-name: "XGBoost"
-backend: "fil"
+backend: "fil",
 max_batch_size: 0
+default_model_filename: "xgboost.json"
 
-input [
+input [                                 
+  {  
+    name: "input__0"
+    data_type: TYPE_FP32
+    dims: [ 3 ]
+  }
 ]
 
 output [
-]
-
-
-
-
-instance_group [
   {
-    kind: KIND_AUTO
-    count: 1
+    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" }
   }
 ]