tfx_bsl.public.tfxio.RecordBatchToExamplesEncoder

Stay organized with collections Save and categorize content based on your preferences.

Encodes pa.RecordBatch as a list of serialized tf.Examples.

tfx_bsl.public.tfxio.RecordBatchToExamplesEncoder(
    schema: Optional[schema_pb2.Schema] = None
)

Requires TFMD schema only if RecordBatches contains nested lists with depth > 2 that represent TensorFlow's RaggedFeatures.

Methods

encode

encode(
    record_batch: pa.RecordBatch
) -> List[bytes]