Filters extracts to include/exclude specified keys.
tfma.extractors.Filter(
include: Optional[Union[Iterable[str], Dict[str, Any]]] = None,
exclude: Optional[Union[Iterable[str], Dict[str, Any]]] = None
) -> beam.pvalue.PCollection
Args |
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': {} }.
|
Returns |
Filtered PCollection of Extracts.
|
Raises |
ValueError
|
If both include and exclude are used.
|