Skip to content

Commit

Permalink
Deprecare boosted trees code in Estimator
Browse files Browse the repository at this point in the history
PiperOrigin-RevId: 413728388
  • Loading branch information
mihaimaruseac authored and tensorflower-gardener committed Dec 2, 2021
1 parent e40893d commit edc9c24
Showing 1 changed file with 4 additions and 0 deletions.
4 changes: 4 additions & 0 deletions tensorflow_estimator/python/estimator/canned/boosted_trees.py
Original file line number Diff line number Diff line change
Expand Up @@ -34,6 +34,7 @@
from tensorflow.python.ops import cond_v2
from tensorflow.python.ops import lookup_ops
from tensorflow.python.ops.array_ops import identity as tf_identity
from tensorflow.python.util import deprecation
from tensorflow.python.util.tf_export import estimator_export
from tensorflow_estimator.python.estimator import estimator
from tensorflow_estimator.python.estimator.canned import boosted_trees_utils
Expand Down Expand Up @@ -1929,6 +1930,7 @@ def _validate_input_params(tree_params):

# pylint: disable=protected-access
@estimator_export('estimator.BoostedTreesClassifier')
@deprecation.deprecated(None, 'Boosted Trees kernels in TF are deprecated. Please use TensorFlow Decision Forests instead (https://github.com/tensorflow/decision-forests)')
class BoostedTreesClassifier(_BoostedTreesBase):
"""A Classifier for Tensorflow Boosted Trees models.
Expand Down Expand Up @@ -2102,6 +2104,7 @@ def _model_fn(features, labels, mode, config):


@estimator_export('estimator.BoostedTreesRegressor')
@deprecation.deprecated(None, 'Boosted Trees kernels in TF are deprecated. Please use TensorFlow Decision Forests instead (https://github.com/tensorflow/decision-forests)')
class BoostedTreesRegressor(_BoostedTreesBase):
"""A Regressor for Tensorflow Boosted Trees models.
Expand Down Expand Up @@ -2253,6 +2256,7 @@ def _model_fn(features, labels, mode, config):


@estimator_export('estimator.BoostedTreesEstimator')
@deprecation.deprecated(None, 'Boosted Trees kernels in TF are deprecated. Please use TensorFlow Decision Forests instead (https://github.com/tensorflow/decision-forests)')
class BoostedTreesEstimator(_BoostedTreesBase): # pylint: disable=protected-access
"""An Estimator for Tensorflow Boosted Trees models."""

Expand Down

0 comments on commit edc9c24

Please sign in to comment.