experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fixing flaky test in quantile regression

Browse Source 
PiperOrigin-RevId: 228376536
This commit is contained in:
A. Unique TensorFlower 2019-01-08 12:18:59 -08:00 committed by TensorFlower Gardener
parent 989c8b40a9
commit 67d8a0330b
1 changed files with 8 additions and 8 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

16
tensorflow/contrib/boosted_trees/estimator_batch/estimator_test.py
View File

@ -487,8 +487,8 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
self.assertTrue(frac_below_upper_0 <= 0.98)
self.assertTrue(frac_below_upper_1 >= 0.92)
self.assertTrue(frac_below_upper_1 <= 0.98)
self.assertTrue(frac_both_below_upper >= 0.92)
self.assertTrue(frac_both_below_upper <= 0.98)
self.assertTrue(frac_both_below_upper >= 0.91)
self.assertTrue(frac_both_below_upper <= 0.99)
train_input_fn, test_input_fn, _ = _quantile_regression_input_fns(
two_dimension=True)
@ -516,8 +516,8 @@ class BoostedTreeEstimatorTest(test_util.TensorFlowTestCase):
self.assertTrue(frac_above_lower_0 <= 0.98)
self.assertTrue(frac_above_lower_1 >= 0.92)
self.assertTrue(frac_above_lower_1 <= 0.98)
self.assertTrue(frac_both_above_lower >= 0.92)
self.assertTrue(frac_both_above_lower <= 0.98)
self.assertTrue(frac_both_above_lower >= 0.91)
self.assertTrue(frac_both_above_lower <= 0.99)
class CoreGradientBoostedDecisionTreeEstimators(test_util.TensorFlowTestCase):
@ -806,8 +806,8 @@ class CoreGradientBoostedDecisionTreeEstimators(test_util.TensorFlowTestCase):
self.assertTrue(frac_below_upper_0 <= 0.98)
self.assertTrue(frac_below_upper_1 >= 0.92)
self.assertTrue(frac_below_upper_1 <= 0.98)
self.assertTrue(frac_both_below_upper >= 0.92)
self.assertTrue(frac_both_below_upper <= 0.98)
self.assertTrue(frac_both_below_upper >= 0.91)
self.assertTrue(frac_both_below_upper <= 0.99)
train_input_fn, test_input_fn, _ = _quantile_regression_input_fns(
two_dimension=True)
@ -835,8 +835,8 @@ class CoreGradientBoostedDecisionTreeEstimators(test_util.TensorFlowTestCase):
self.assertTrue(frac_above_lower_0 <= 0.98)
self.assertTrue(frac_above_lower_1 >= 0.92)
self.assertTrue(frac_above_lower_1 <= 0.98)
self.assertTrue(frac_both_above_lower >= 0.92)
self.assertTrue(frac_both_above_lower <= 0.98)
self.assertTrue(frac_both_above_lower >= 0.91)
self.assertTrue(frac_both_above_lower <= 0.99)
if __name__ == "__main__":