View source on GitHub |
Creates an extractor for extracting slice keys.
tfma.extractors.SliceKeyExtractor(
slice_spec: Optional[List[slicer.SingleSliceSpec]] = None,
eval_config: Optional[tfma.EvalConfig
] = None,
materialize: Optional[bool] = True
) -> tfma.extractors.Extractor
The incoming Extracts must contain features stored under tfma.FEATURES_KEY and optionally under tfma.TRANSFORMED_FEATURES.
The extractor's PTransform yields a copy of the Extracts input with an additional extract pointing at the list of SliceKeyType values keyed by tfma.SLICE_KEY_TYPES_KEY. If materialize is True then a materialized version of the slice keys will be added under the key tfma.SLICE_KEYS_KEY.
Returns | |
---|---|
Extractor for slice keys. |