Add back most of the proto

2020-12-30 17:32:21 -08:00 · 2020-12-30 17:32:21 -08:00 · 251e710762
commit 251e710762
parent de3de808b6
1 changed files with 23 additions and 0 deletions
--- a/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py
+++ b/tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py
@ -106,6 +106,29 @@ class ResourceOpsTest(test_util.TensorFlowTestCase):
            }
          }
        }
+        trees {
+          nodes {
+            bucketized_split {
+              feature_id: 75
+              threshold: 21
+              left_id: 1
+              right_id: 2
+            }
+            metadata {
+              gain: -1.4
+            }
+          }
+          nodes {
+            leaf {
+              scalar: -0.6
+            }
+          }
+          nodes {
+            leaf {
+              scalar: 0.165
+            }
+          }
+        }
      """, ensemble_proto)
      ensemble = boosted_trees_ops.TreeEnsemble(
          'ensemble',