Skip to content

Instantly share code, notes, and snippets.

@leechanwoo

leechanwoo/dataset_class.py

Created August 24, 2017 02:08
Show Gist options
  • Save leechanwoo/89250e96b842c37e390db91e0ad6bfce to your computer and use it in GitHub Desktop.
Save leechanwoo/89250e96b842c37e390db91e0ad6bfce to your computer and use it in GitHub Desktop.
Download ZIP
csv reader simple
Raw
dataset_class.py
dataset = tf.contrib.data.TextLineDataset(labelname).batch(10)
itr = dataset.make_one_shot_iterator()
batch = itr.get_next()
with tf.Session() as sess:
print(sess.run(batch))
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment