tfma.extractors.Filter

View source on GitHub

Filters extracts to include/exclude specified keys.

extracts PCollection of extracts.
include List or map of keys to include in output. If a map of keys is passed then the keys and sub-keys that exist in the map will be included in the output. An empty dict behaves as a wildcard matching all keys or the value itself. Since matching on feature values is not currently supported, an empty dict must be used to represent the leaf nodes. For example: {'key1': {'key1-subkey': {} }, 'key2': {} }.
exclude List or map of keys to exclude from output. If a map of keys is passed then the keys and sub-keys that exist in the map will be excluded from the output. An empty dict behaves as a wildcard matching all keys or the value itself. Since matching on feature values is not currently supported, an empty dict must be used to represent the leaf nodes. For example: {'key1': {'key1-subkey': {} }, 'key2': {} }.

Filtered PCollection of Extracts.

ValueError If both include and exclude are used.