aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2016-10-17 06:24:51 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-10-17 07:35:40 -0700
commit64d29a95e41113d4abf7e64cd1f6d8cdb01bc018 (patch)
tree33eb4bcf58d39abfbdbcab1cddab6becb33a88e2 /tensorflow
parent5b215de4ecd642f792bb6b6e75bbeefee01904b0 (diff)
Mark constructors for RNN Estimators as @experimental.
Change: 136348245
Diffstat (limited to 'tensorflow')
-rw-r--r--tensorflow/contrib/learn/python/learn/estimators/dynamic_rnn_estimator.py5
1 files changed, 5 insertions, 0 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/dynamic_rnn_estimator.py b/tensorflow/contrib/learn/python/learn/estimators/dynamic_rnn_estimator.py
index 249ec2ca39..241b2b41e5 100644
--- a/tensorflow/contrib/learn/python/learn/estimators/dynamic_rnn_estimator.py
+++ b/tensorflow/contrib/learn/python/learn/estimators/dynamic_rnn_estimator.py
@@ -23,6 +23,7 @@ import six
from tensorflow.contrib import framework as contrib_framework
from tensorflow.contrib import layers
+from tensorflow.contrib.framework.python.framework import experimental
from tensorflow.contrib.learn.python.learn.estimators import estimator
from tensorflow.python.framework import dtypes
from tensorflow.python.framework import ops
@@ -544,6 +545,7 @@ def _get_rnn_cell(cell_type, num_units, num_layers):
return cell
+@experimental
def multi_value_rnn_regressor(num_units,
sequence_feature_columns,
context_feature_columns=None,
@@ -596,6 +598,7 @@ def multi_value_rnn_regressor(num_units,
dtype=cell_dtype)
+@experimental
def multi_value_rnn_classifier(num_classes,
num_units,
sequence_feature_columns,
@@ -650,6 +653,7 @@ def multi_value_rnn_classifier(num_classes,
dtype=cell_dtype)
+@experimental
def single_value_rnn_regressor(num_units,
sequence_feature_columns,
context_feature_columns=None,
@@ -702,6 +706,7 @@ def single_value_rnn_regressor(num_units,
dtype=cell_dtype)
+@experimental
def single_value_rnn_classifier(num_classes,
num_units,
sequence_feature_columns,