Upload config.pbtxt

2025-09-29 06:27:56 +00:00 · 2025-09-29 06:27:56 +00:00 · 3a522b6d89
commit 3a522b6d89
parent 96057ff406
1 changed files with 46 additions and 0 deletions
--- a/config.pbtxt
+++ b/config.pbtxt
@ -0,0 +1,46 @@
+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" }
+  }
+]
+