Fixing flake in the test.

PiperOrigin-RevId: 262346360
This commit is contained in:
A. Unique TensorFlower 2019-08-08 07:20:26 -07:00 committed by TensorFlower Gardener
parent e522f75336
commit 3cf761fdd9

View File

@ -501,7 +501,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
}
nodes {
dense_float_binary_split {
feature_column: 1
feature_column: 0
threshold: 0.52
left_id: 3
right_id: 4
@ -512,7 +512,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
}
nodes {
dense_float_binary_split {
feature_column: 1
feature_column: 0
threshold: 0.554
left_id: 5
right_id: 6
@ -589,8 +589,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
}
nodes {
dense_float_binary_split {
feature_column: 1
threshold: 0.52
threshold: 0.519999980927
left_id: 3
right_id: 4
}
@ -599,8 +598,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
}
nodes {
dense_float_binary_split {
feature_column: 1
threshold: 0.554
threshold: 0.554000020027
left_id: 5
right_id: 6
}
@ -621,6 +619,13 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
}
}
}
nodes {
leaf {
vector {
value: 0.0
}
}
}
nodes {
dense_float_binary_split {
threshold: 1.0
@ -631,13 +636,6 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
gain: 0.888888895512
}
}
nodes {
leaf {
vector {
value: 0.0
}
}
}
nodes {
leaf {
vector {
@ -665,8 +663,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
}
nodes {
dense_float_binary_split {
feature_column: 1
threshold: 0.52
threshold: 0.519999980927
left_id: 3
right_id: 4
}
@ -675,8 +672,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
}
nodes {
dense_float_binary_split {
feature_column: 1
threshold: 0.554
threshold: 0.554000020027
left_id: 5
right_id: 6
}
@ -697,6 +693,13 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
}
}
}
nodes {
leaf {
vector {
value: 0.0
}
}
}
nodes {
dense_float_binary_split {
threshold: 1.0
@ -707,13 +710,6 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
gain: 0.727760672569
}
}
nodes {
leaf {
vector {
value: 0.0
}
}
}
nodes {
leaf {
vector {
@ -741,8 +737,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
}
nodes {
dense_float_binary_split {
feature_column: 1
threshold: 0.52
threshold: 0.519999980927
left_id: 3
right_id: 4
}
@ -751,8 +746,7 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
}
nodes {
dense_float_binary_split {
feature_column: 1
threshold: 0.554
threshold: 0.554000020027
left_id: 5
right_id: 6
}