aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/examples/skflow/multioutput_regression.py
diff options
context:
space:
mode:
authorGravatar Wei Ho <weiho@google.com>2016-10-17 10:02:36 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-10-17 12:47:04 -0700
commit0b57f101c25b5c0db0f9ecf9af7320faf2821d0e (patch)
tree0b45206f3ae2370975f81b3cb72230d5c891b3c7 /tensorflow/examples/skflow/multioutput_regression.py
parent40aee90b9217a5f72360bcf25cc1a0999ff5d464 (diff)
Delete dnn_ops.py
Change: 136370181
Diffstat (limited to 'tensorflow/examples/skflow/multioutput_regression.py')
-rw-r--r--tensorflow/examples/skflow/multioutput_regression.py9
1 files changed, 6 insertions, 3 deletions
diff --git a/tensorflow/examples/skflow/multioutput_regression.py b/tensorflow/examples/skflow/multioutput_regression.py
index 8a22346d4d..63007118b2 100644
--- a/tensorflow/examples/skflow/multioutput_regression.py
+++ b/tensorflow/examples/skflow/multioutput_regression.py
@@ -21,9 +21,11 @@ from __future__ import absolute_import
from __future__ import division
from __future__ import print_function
-import numpy as np
+
import matplotlib.pyplot as plt
+import numpy as np
from sklearn.metrics import mean_squared_error
+import tensorflow as tf
from tensorflow.contrib import learn
@@ -37,8 +39,9 @@ regressors = []
options = [[2], [10, 10], [20, 20]]
for hidden_units in options:
def tanh_dnn(X, y):
- features = learn.ops.dnn(X, hidden_units=hidden_units,
- activation=learn.tf.tanh)
+ dnn = lambda inputs, num_outputs, scope: tf.contrib.layers.legacy_fully_connected(
+ inputs, num_outputs, weight_init=None, activation_fn=learn.tf.tanh)
+ features = tf.contrib.layers.stack(X, dnn, hidden_units)
return learn.models.linear_regression(features, y)
regressor = learn.TensorFlowEstimator(model_fn=tanh_dnn, n_classes=0,