tfma.model_util.rebatch_by_input_names

View source on GitHub

Converts a batch of extracts into multiple batches keyed by input names.

batch_of_extracts Batch of extracts (one per example).
input_names List of input names to search for features under.
input_specs Optional list of type specs associated with inputs.

Dict of batch aligned features keyed by input (feature) name.