Module: tfma.extractors

Defined in extractors/__init__.py.

Init module for TensorFlow Model Analysis extractors.

Classes

class Extractor: Extractor(stage_name, ptransform)

Functions

FeatureExtractor(...)

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

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

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

Other Members

FEATURE_EXTRACTOR_STAGE_NAME

LAST_EXTRACTOR_STAGE_NAME

PREDICT_EXTRACTOR_STAGE_NAME

SLICE_KEY_EXTRACTOR_STAGE_NAME