tfdf.py_tree.dataspec.categorical_vocab_iterator

Returns a categorical spec's vocabulary as a list.

tfdf.py_tree.dataspec.categorical_vocab_iterator(
    categorical_spec: data_spec_pb2.CategoricalSpec
) -> Iterator[Tuple[bytes, data_spec_pb2.CategoricalSpec.VocabValue]]

The data spec currently encodes the vocabulary as a map from string to VocabValue. If a key of this map is of type bytes, protobuf's python implementation may fail to retrieve the values from this map. This is a helper function to work around this limitation.

Note that this function is internal and may change or be removed at any time.

categorical_spec A categorical spec.

The vocabulary as (key, value) tuples.