View source on GitHub |
Creates an extractor for unbatching batched extracts.
tfma.extractors.UnbatchExtractor() -> tfma.extractors.Extractor
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).
Returns | |
---|---|
Extractor for unbatching batched extracts. |