Skip to content

GitLab

"vscode:/vscode.git/clone" did not exist on "5b1fc42a82422a82b3603ba272894d3d08ff2406"
Commit 2ed6b9d9 authored by Neal Wu's avatar Neal Wu
Browse files

Fix wide_deep_test.py

parent e8a7f352
Show whitespace changes
Inline Side-by-side
Showing with 3 additions and 1 deletion
official/wide_deep/wide_deep_test.py
View file @ 2ed6b9d9
...@@ -54,7 +54,9 @@ class BaseTest(tf.test.TestCase): ...@@ -54,7 +54,9 @@ class BaseTest(tf.test.TestCase):
temp_csv.write(TEST_INPUT) temp_csv.write(TEST_INPUT)
def test_input_fn(self): def test_input_fn(self):
features, labels = wide_deep.input_fn(self.input_csv, 1, False, 1) dataset = wide_deep.input_fn(self.input_csv, 1, False, 1)
features, labels = dataset.make_one_shot_iterator().get_next()
with tf.Session() as sess: with tf.Session() as sess:
features, labels = sess.run((features, labels)) features, labels = sess.run((features, labels))
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment