diff options
Diffstat (limited to 'tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py')
-rw-r--r-- | tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py b/tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py index d1113678a9..f472034225 100644 --- a/tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py +++ b/tensorflow/contrib/learn/python/learn/estimators/dnn_linear_combined.py @@ -171,7 +171,7 @@ def _dnn_linear_combined_model_fn(features, labels, mode, params, config=None): dnn_feature_columns = params.get("dnn_feature_columns") dnn_optimizer = params.get("dnn_optimizer") or "Adagrad" dnn_hidden_units = params.get("dnn_hidden_units") - dnn_activation_fn = params.get("dnn_activation_fn") + dnn_activation_fn = params.get("dnn_activation_fn") or nn.relu dnn_dropout = params.get("dnn_dropout") gradient_clip_norm = params.get("gradient_clip_norm") input_layer_min_slice_size = ( @@ -346,7 +346,7 @@ class _DNNLinearCombinedEstimator(estimator.Estimator): dnn_feature_columns=None, dnn_optimizer=None, dnn_hidden_units=None, - dnn_activation_fn=nn.relu, + dnn_activation_fn=None, dnn_dropout=None, gradient_clip_norm=None, config=None, |