TensorFlow 2 version | 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
close
close()
Close the file.
flush
flush()
Flush the file.
write
write(
record
)
Write a string record to the file.
Args | |
---|---|
record
|
str |
__enter__
__enter__()
Enter a with
block.
__exit__
__exit__(
unused_type, unused_value, unused_traceback
)
Exit a with
block, closing the file.