Skip to content

GitLab

"...text-generation-inference.git" did not exist on "6ec5288ab76ce91e84bd7389a0249b0254d1d1f8"
Commit 2ed6b9d9 authored by Neal Wu's avatar Neal Wu
Browse files

Fix wide_deep_test.py

parent e8a7f352
Hide 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