Module: tfma.extractors

Init module for TensorFlow Model Analysis extractors.

Classes

class Extractor: Extractor(stage_name, ptransform)

Functions

ExampleWeightsExtractor(...): Creates an extractor for extracting example weights.

FeatureExtractor(...)

FeaturesExtractor(...): Creates an extractor for extracting features.

Filter(...): Filters extracts to include/exclude specified keys.

InputExtractor(...): Creates an extractor for extracting features, labels, and example weights.

LabelsExtractor(...): Creates an extractor for extracting labels.

PredictExtractor(...): Creates an Extractor for TFMAPredict.

PredictionsExtractor(...): Creates an extractor for performing predictions over a batch.

SliceKeyExtractor(...): Creates an extractor for extracting slice keys.

SLICE_KEY_EXTRACTOR_STAGE_NAME 'ExtractSliceKeys'