tfma.extractors.UnbatchExtractor

Creates an extractor for unbatching batched extracts.

This extractor removes Arrow RecordBatch from the batched extract and outputs per-example extracts with the remaining keys. We assume that the remaining keys in the input extract contain list of objects (one per example).

Extractor for unbatching batched extracts.