View source on GitHub
|
Init module for TensorFlow Model Analysis extractors.
Classes
class Extractor: Extractor(stage_name, ptransform)
Functions
ExampleWeightsExtractor(...): Creates an extractor for extracting example weights.
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.
TransformedFeaturesExtractor(...): Creates an extractor for extracting transformed features.
UnbatchExtractor(...): Creates an extractor for unbatching batched extracts.
Other Members | |
|---|---|
| SLICE_KEY_EXTRACTOR_STAGE_NAME |
'ExtractSliceKeys'
|