projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: e7a9508) | patch
enhance Tensorflow GBDT and GBRT model by exposing a new two dimensional output in...
authorA. Unique TensorFlower <gardener@tensorflow.org>
Fri, 25 May 2018 19:35:50 +0000 (12:35 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Fri, 25 May 2018 19:38:35 +0000 (12:38 -0700)
commitae0eb1b7f81f6d98e0503b9568c72feaa805e655
tree806a14c93809ca9277d7e7d459c64e0435c3cdfdtree | snapshot
parente7a9508fe3447bf51cebad48a86e1846a5a48d70commit | diff
enhance Tensorflow GBDT and GBRT model by exposing a new two dimensional output in prediction ops (example id, tree leaf node index id) for input as other model features

PiperOrigin-RevId: 198087342
tensorflow/contrib/boosted_trees/estimator_batch/estimator.py diff | blob | history
tensorflow/contrib/boosted_trees/estimator_batch/estimator_test.py diff | blob | history
tensorflow/contrib/boosted_trees/estimator_batch/model.py diff | blob | history
tensorflow/contrib/boosted_trees/kernels/prediction_ops.cc diff | blob | history
tensorflow/contrib/boosted_trees/lib/models/multiple_additive_trees.cc diff | blob | history
tensorflow/contrib/boosted_trees/lib/models/multiple_additive_trees.h diff | blob | history
tensorflow/contrib/boosted_trees/lib/models/multiple_additive_trees_test.cc diff | blob | history
tensorflow/contrib/boosted_trees/ops/prediction_ops.cc diff | blob | history
tensorflow/contrib/boosted_trees/python/ops/prediction_ops.py diff | blob | history
tensorflow/contrib/boosted_trees/python/training/functions/gbdt_batch.py diff | blob | history
tensorflow/contrib/boosted_trees/python/training/functions/gbdt_batch_test.py diff | blob | history
Domain: Machine Learning / ML Framework;