tf.io.TFRecordWriter

TensorFlow 1 version View source on GitHub

A class to write records to a TFRecords file.

tf.io.TFRecordWriter(
    path, options=None
)

Used in the notebooks

Used in the tutorials

This class implements __enter__ and __exit__, and can be used in with blocks like a normal file.

Args:

  • path: The path to the TFRecords file.
  • options: (optional) String specifying compression type, TFRecordCompressionType, or TFRecordOptions object.

Raises:

  • IOError: If path cannot be opened for writing.
  • ValueError: If valid compression_type can't be determined from options.

Methods

__enter__

View source

__enter__()

Enter a with block.

__exit__

View source

__exit__(
    unused_type, unused_value, unused_traceback
)

Exit a with block, closing the file.

close

View source

close()

Close the file.

flush

View source

flush()

Flush the file.

write

View source

write(
    record
)

Write a string record to the file.

Args:

  • record: str