diff --git a/config.pbtxt b/config.pbtxt new file mode 100644 index 0000000..a331400 --- /dev/null +++ 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" } + } +] +