Add model dups for FVT (#10)

Towards adding a couple of tests in kserve/modelmesh-serving#485 Signed-off-by: Rafael Vasquez <[email protected]>
kserve · Mar 13, 2024 · 765d10b · 765d10b
1 parent 09ed9c9
commit 765d10b
Show file tree

Hide file tree

Showing 6 changed files with 2,146 additions and 0 deletions.
diff --git a/fvt/lightgbm/lightgbm-fil/1/model.txt → fvt/lightgbm/mushroom-fil-dup/1/model.txt b/fvt/lightgbm/lightgbm-fil/1/model.txt → fvt/lightgbm/mushroom-fil-dup/1/model.txt
diff --git a/fvt/lightgbm/lightgbm-fil/config.pbtxt → fvt/lightgbm/mushroom-fil-dup/config.pbtxt b/fvt/lightgbm/lightgbm-fil/config.pbtxt → fvt/lightgbm/mushroom-fil-dup/config.pbtxt
diff --git a/fvt/lightgbm/mushroom-fil/1/model.txt b/fvt/lightgbm/mushroom-fil/1/model.txt
diff --git a/fvt/lightgbm/mushroom-fil/config.pbtxt b/fvt/lightgbm/mushroom-fil/config.pbtxt
@@ -0,0 +1,27 @@
+backend: "fil"
+max_batch_size: 8
+input [
+ {
+    name: "input__0"
+    data_type: TYPE_FP32
+    dims: [ 126 ]
+  }
+]
+output [
+ {
+    name: "output__0"
+    data_type: TYPE_FP32
+    dims: [ 1 ]
+  }
+]
+instance_group [{ kind: KIND_AUTO }]
+parameters [
+  {
+    key: "model_type"
+    value: { string_value: "lightgbm" }
+  },
+  {
+    key: "output_class"
+    value: { string_value: "false" }
+  }
+]
diff --git a/fvt/xgboost/mushroom-fil-dup/1/xgboost.json b/fvt/xgboost/mushroom-fil-dup/1/xgboost.json
@@ -0,0 +1,87 @@
+{
+  "learner": {
+    "attributes": {},
+    "gradient_booster": {
+      "model": {
+        "gbtree_model_param": {"num_trees": "2", "size_leaf_vector": "0"},
+        "tree_info": [0, 0],
+        "trees": [
+          {
+            "base_weights": [
+              -7.15052932500839233e-2, 1.29551589488983154,
+              -1.86661934852600098, 1.71217715740203857, -1.70044052600860596,
+              -1.94070863723754883, 1.8596491813659668
+            ],
+            "default_left": [true, true, true, false, false, false, false],
+            "id": 0,
+            "leaf_child_counts": [0, 0, 0, 0, 0, 0, 0],
+            "left_children": [1, 3, 5, -1, -1, -1, -1],
+            "loss_changes": [
+              4.000531005859375e3, 1.1582120361328125e3, 1.98173828125e2, 0.0,
+              0.0, 0.0, 0.0
+            ],
+            "parents": [2147483647, 0, 0, 1, 1, 2, 2],
+            "right_children": [2, 4, 6, -1, -1, -1, -1],
+            "split_conditions": [
+              -9.5367431640625e-7, -9.5367431640625e-7, -9.5367431640625e-7,
+              1.71217715740203857, -1.70044052600860596, -1.94070863723754883,
+              1.8596491813659668
+            ],
+            "split_indices": [28, 55, 108, 0, 0, 0, 0],
+            "sum_hessian": [
+              1.62825e3, 9.245e2, 7.0375e2, 8.12e2, 1.125e2, 6.905e2, 1.325e1
+            ],
+            "tree_param": {
+              "num_deleted": "0",
+              "num_feature": "126",
+              "num_nodes": "7",
+              "size_leaf_vector": "0"
+            }
+          },
+          {
+            "base_weights": [
+              -9.31910648941993713e-2, 7.82926455140113831e-2,
+              -6.23624467849731445, 7.84717559814453125e-1,
+              -9.68530356884002686e-1
+            ],
+            "default_left": [true, true, false, false, false],
+            "id": 1,
+            "leaf_child_counts": [0, 0, 0, 0, 0],
+            "left_children": [1, 3, -1, -1, -1],
+            "loss_changes": [
+              8.325450439453125e2, 5.6972509765625e2, 0.0, 0.0, 0.0
+            ],
+            "parents": [2147483647, 0, 0, 1, 1],
+            "right_children": [2, 4, -1, -1, -1],
+            "split_conditions": [
+              -9.5367431640625e-7, -9.5367431640625e-7, -6.23624467849731445,
+              7.84717559814453125e-1, -9.68530356884002686e-1
+            ],
+            "split_indices": [59, 28, 0, 0, 0],
+            "sum_hessian": [
+              7.8885205078125e2, 7.6838970947265625e2, 2.04623889923095703e1,
+              4.58936859130859375e2, 3.0945281982421875e2
+            ],
+            "tree_param": {
+              "num_deleted": "0",
+              "num_feature": "126",
+              "num_nodes": "5",
+              "size_leaf_vector": "0"
+            }
+          }
+        ]
+      },
+      "name": "gbtree"
+    },
+    "learner_model_param": {
+      "base_score": "0.500000",
+      "num_class": "0",
+      "num_feature": "126"
+    },
+    "objective": {
+      "name": "binary:logistic",
+      "reg_loss_param": {"scale_pos_weight": "1"}
+    }
+  },
+  "version": [1, 1, 1]
+}
diff --git a/fvt/xgboost/mushroom-fil-dup/config.pbtxt b/fvt/xgboost/mushroom-fil-dup/config.pbtxt
@@ -0,0 +1,27 @@
+backend: "fil"
+max_batch_size: 8
+input [
+ {
+    name: "input__0"
+    data_type: TYPE_FP32
+    dims: [ 126 ]
+  }
+]
+output [
+ {
+    name: "output__0"
+    data_type: TYPE_FP32
+    dims: [ 1 ]
+  }
+]
+instance_group [{ kind: KIND_AUTO }]
+parameters [
+  {
+    key: "model_type"
+    value: { string_value: "xgboost_json" }
+  },
+  {
+    key: "output_class"
+    value: { string_value: "false" }
+  }
+]