tf.io.TFRecordWriter

View source on GitHub

A class to write records to a TFRecords file.

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

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