ML Community Day is November 9! Join us for updates from TensorFlow, JAX, and more Learn more

tfma.extractors.FeaturesExtractor

Creates an extractor for extracting features.

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.

eval_config Eval config.

Extractor for extracting features.