Upload config.pbtxt
This commit is contained in:
parent
96057ff406
commit
3a522b6d89
46
config.pbtxt
Normal file
46
config.pbtxt
Normal file
@ -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" }
|
||||||
|
}
|
||||||
|
]
|
||||||
|
|
||||||
Loading…
Reference in New Issue
Block a user