Module: tfma.extractors

Init module for TensorFlow Model Analysis extractors.

Classes

class Extractor: Extractor(stage_name, ptransform)

Functions

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

FeatureExtractor(...)

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

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

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

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

SLICE_KEY_EXTRACTOR_STAGE_NAME 'ExtractSliceKeys'