Creates an extractor for extracting labels.
tfma.extractors.LabelsExtractor(
eval_config: tfma.EvalConfig
) -> tfma.extractors.Extractor
The extractor's PTransform uses the config's ModelSpec.label_key(s) to lookup
the associated label values stored as features under the tfma.FEATURES_KEY
(and optionally tfma.TRANSFORMED_FEATURES_KEY) in extracts. The resulting
values are then added to the extracts under the key tfma.LABELS_KEY.
Args |
|---|
eval_config
|
Eval config.
|
Returns |
|---|
|
Extractor for extracting labels.
|
View source on GitHub