Module: tfma.extractors

View source on GitHub

Init module for TensorFlow Model Analysis extractors.

Classes

class Extractor: Extractor(stage_name, ptransform)

Functions

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.

Other Members

  • FEATURE_EXTRACTOR_STAGE_NAME = 'ExtractFeatures'
  • LAST_EXTRACTOR_STAGE_NAME = '<last-extractor>'
  • PREDICT_EXTRACTOR_STAGE_NAME = 'Predict'
  • SLICE_KEY_EXTRACTOR_STAGE_NAME = 'ExtractSliceKeys'