boosted_trees: fixed the crash when eval/prediction is attempted with the initial...
authorYounghee Kwon <youngheek@google.com>
Thu, 17 May 2018 23:52:17 +0000 (16:52 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Thu, 17 May 2018 23:58:20 +0000 (16:58 -0700)
commita3b86f9c2dfbe2cff6a3b6362f9e0f131246eb0b
tree7f499dce2fa29e1dfe91196a24be129b89c9607f
parent56ba8444c5b65c932e910e18b1ee3d1c5c4243f6
boosted_trees: fixed the crash when eval/prediction is attempted with the initial checkpoints (at step 0) before training.

PiperOrigin-RevId: 197073582
tensorflow/core/kernels/boosted_trees/prediction_ops.cc
tensorflow/python/estimator/canned/boosted_trees_test.py
tensorflow/python/kernel_tests/boosted_trees/prediction_ops_test.py