Fixing flake in the test.
PiperOrigin-RevId: 262346360
This commit is contained in:
parent
e522f75336
commit
3cf761fdd9
@ -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
|
||||
}
|
||||
|
||||
Loading…
x
Reference in New Issue
Block a user