tfma.utils.merge_extracts

Merges list of extracts into a single extract with multidimensional data.

tfma.utils.merge_extracts(
    extracts: List[tfma.types.Extracts],
    squeeze_two_dim_vector: bool = True
) -> tfma.types.Extracts

Note: Running split_extracts followed by merge extracts with default options will not reproduce the exact shape of the original extracts. Arrays in shape (x,1) will be flattened to (x,). To maintain the original shape of extract values of array shape (x,1), you must run with these options: split_extracts(extracts, expand_zero_dims=False) merge_extracts(extracts, squeeze_two_dim_vector=False) Args: extracts: Batched TFMA Extracts. squeeze_two_dim_vector: Determines how the function will handle arrays of shape (x,1). If squeeze_two_dim_vector is True, the array will be squeezed to shape (x,).

Returns
A single Extracts whose values have been grouped into batches.