Skip to content

Instantly share code, notes, and snippets.

@tgamauf

tgamauf/tf_tfrecordwriter.py

Last active March 4, 2018 12:20
Show Gist options
  • Save tgamauf/61391f93369eb2ed4b2d293b0bd7af5d to your computer and use it in GitHub Desktop.
Save tgamauf/61391f93369eb2ed4b2d293b0bd7af5d to your computer and use it in GitHub Desktop.
Download ZIP
Use of tf.python_io.TFRecordWriter
Raw
tf_tfrecordwriter.py
# "example" is of type tf.train.Example.
with tf.python_io.TFRecordWriter('movie_ratings.tfrecord') as writer:
writer.write(example.SerializeToString())
Sign up for free to join this conversation on GitHub. Already have an account? Sign in to comment