Fill in tree node

2021-01-03 16:35:43 -08:00 · 2021-01-03 16:35:43 -08:00 · 16b995513d
commit 16b995513d
parent dd5f07f812
1 changed files with 2 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
@ -51,6 +51,7 @@ class ResourceOpsTest(test_util.TensorFlowTestCase):
          nodes {
            bucketized_split {
              feature_id: 4
+              threshold: 21
              left_id: 1
              right_id: 2
            }
@ -60,6 +61,7 @@ class ResourceOpsTest(test_util.TensorFlowTestCase):
          }
          nodes {
            bucketized_split {
+              feature_id: 4
              threshold: 21
              left_id: 3
              right_id: 4