temp_csv.write(TEST_INPUT)
def test_input_fn(self):
features, labels = wide_deep.input_fn(self.input_csv, 1, False, 1)()
with tf.Session() as sess:
features, labels = sess.run((features, labels))
// Compare the two features dictionaries.
After Change
temp_csv.write(TEST_INPUT)
def test_input_fn(self):
features, labels = wide_deep.input_fn(self.input_csv, 1, False, 1)
with tf.Session() as sess:
features, labels = sess.run((features, labels))
// Compare the two features dictionaries.