projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: ae3d20f) | patch
Added kernels and estimators for Gradient Boosting Trees algorithm.
authorYounghee Kwon <youngheek@google.com>
Thu, 29 Mar 2018 16:43:19 +0000 (09:43 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Thu, 29 Mar 2018 16:48:47 +0000 (09:48 -0700)
commita2b6c3c124664d682094a1ecfa9cc00cca8ada85
tree1f72bb1a6e6542bfd5e215b535e46b9e782459d5tree | snapshot
parentae3d20f9aef78554f0d0f5eec13982e9802a45d2commit | diff
Added kernels and estimators for Gradient Boosting Trees algorithm.
BoostedTreesClassifier and BoostedTreesRegressor are added to tf.estimator.
Also some training utility functions are added to tf.contrib.estimator.

PiperOrigin-RevId: 190942599
49 files changed:
tensorflow/contrib/cmake/python_modules.txt diff | blob | history
tensorflow/contrib/cmake/python_protos.txt diff | blob | history
tensorflow/contrib/cmake/tf_core_ops.cmake diff | blob | history
tensorflow/contrib/cmake/tf_python.cmake diff | blob | history
tensorflow/contrib/estimator/BUILD diff | blob | history
tensorflow/contrib/estimator/__init__.py diff | blob | history
tensorflow/contrib/estimator/python/estimator/boosted_trees.py [new file with mode: 0644] blob
tensorflow/contrib/estimator/python/estimator/boosted_trees_test.py [new file with mode: 0644] blob
tensorflow/contrib/makefile/tf_op_files.txt diff | blob | history
tensorflow/contrib/makefile/tf_proto_files.txt diff | blob | history
tensorflow/core/BUILD diff | blob | history
tensorflow/core/api_def/base_api/api_def_BoostedTreesCalculateBestGainsPerFeature.pbtxt [new file with mode: 0644] blob
tensorflow/core/api_def/base_api/api_def_BoostedTreesCreateEnsemble.pbtxt [new file with mode: 0644] blob
tensorflow/core/api_def/base_api/api_def_BoostedTreesDeserializeEnsemble.pbtxt [new file with mode: 0644] blob
tensorflow/core/api_def/base_api/api_def_BoostedTreesEnsembleResourceHandleOp.pbtxt [new file with mode: 0644] blob
tensorflow/core/api_def/base_api/api_def_BoostedTreesGetEnsembleStates.pbtxt [new file with mode: 0644] blob
tensorflow/core/api_def/base_api/api_def_BoostedTreesMakeStatsSummary.pbtxt [new file with mode: 0644] blob
tensorflow/core/api_def/base_api/api_def_BoostedTreesPredict.pbtxt [new file with mode: 0644] blob
tensorflow/core/api_def/base_api/api_def_BoostedTreesSerializeEnsemble.pbtxt [new file with mode: 0644] blob
tensorflow/core/api_def/base_api/api_def_BoostedTreesTrainingPredict.pbtxt [new file with mode: 0644] blob
tensorflow/core/api_def/base_api/api_def_BoostedTreesUpdateEnsemble.pbtxt [new file with mode: 0644] blob
tensorflow/core/api_def/base_api/api_def_IsBoostedTreesEnsembleInitialized.pbtxt [new file with mode: 0644] blob
tensorflow/core/kernels/BUILD diff | blob | history
tensorflow/core/kernels/boosted_trees/BUILD [new file with mode: 0644] blob
tensorflow/core/kernels/boosted_trees/boosted_trees.proto [new file with mode: 0644] blob
tensorflow/core/kernels/boosted_trees/prediction_ops.cc [new file with mode: 0644] blob
tensorflow/core/kernels/boosted_trees/resource_ops.cc [new file with mode: 0644] blob
tensorflow/core/kernels/boosted_trees/resources.cc [new file with mode: 0644] blob
tensorflow/core/kernels/boosted_trees/resources.h [new file with mode: 0644] blob
tensorflow/core/kernels/boosted_trees/stats_ops.cc [new file with mode: 0644] blob
tensorflow/core/kernels/boosted_trees/training_ops.cc [new file with mode: 0644] blob
tensorflow/core/ops/boosted_trees_ops.cc [new file with mode: 0644] blob
tensorflow/python/BUILD diff | blob | history
tensorflow/python/__init__.py diff | blob | history
tensorflow/python/estimator/BUILD diff | blob | history
tensorflow/python/estimator/canned/boosted_trees.py [new file with mode: 0644] blob
tensorflow/python/estimator/canned/boosted_trees_test.py [new file with mode: 0644] blob
tensorflow/python/estimator/estimator_lib.py diff | blob | history
tensorflow/python/kernel_tests/boosted_trees/BUILD [new file with mode: 0644] blob
tensorflow/python/kernel_tests/boosted_trees/__init__.py [new file with mode: 0644] blob
tensorflow/python/kernel_tests/boosted_trees/prediction_ops_test.py [new file with mode: 0644] blob
tensorflow/python/kernel_tests/boosted_trees/resource_ops_test.py [new file with mode: 0644] blob
tensorflow/python/kernel_tests/boosted_trees/stats_ops_test.py [new file with mode: 0644] blob
tensorflow/python/kernel_tests/boosted_trees/training_ops_test.py [new file with mode: 0644] blob
tensorflow/python/ops/boosted_trees_ops.py [new file with mode: 0644] blob
tensorflow/python/training/device_setter.py diff | blob | history
tensorflow/tools/api/golden/tensorflow.estimator.-boosted-trees-classifier.pbtxt [new file with mode: 0644] blob
tensorflow/tools/api/golden/tensorflow.estimator.-boosted-trees-regressor.pbtxt [new file with mode: 0644] blob
tensorflow/tools/api/golden/tensorflow.estimator.pbtxt diff | blob | history
Domain: Machine Learning / ML Framework;