Creates an extractor for extracting features.
tfma.extractors.FeaturesExtractor(
eval_config: tfma.EvalConfig
) -> tfma.extractors.Extractor
The extractor's PTransform extracts features from an Arrow RecordBatch stored
under tfma.ARROW_RECORD_BATCH_KEY in the incoming extract and adds them to the
output extract under the key tfma.FEATURES_KEY. Any extracts that already
exist will be merged with the values parsed by this extractor with this
extractor's values taking precedence when duplicate keys are detected.
Args |
eval_config
|
Eval config.
|
Returns |
Extractor for extracting features.
|