|View source on GitHub|
Creates an Extractor for TFMAPredict.
tfma.types.MaybeMultipleEvalSharedModels, desired_batch_size: Optional[int] = None, materialize: Optional[bool] = True, eval_config: Optional[
tfma.EvalConfig] = None ) ->
The extractor's PTransform loads and runs the eval_saved_model against every example yielding a copy of the Extracts input with an additional extract of type FeaturesPredictionsLabels keyed by tfma.FEATURES_PREDICTIONS_LABELS_KEY unless eval_config is not None in which case the features, predictions, and labels will be stored separately under tfma.FEATURES_KEY, tfma.PREDICTIONS_KEY, and tfma.LABELS_KEY respectively.
|Extractor for extracting features, predictions, labels, and other tensors during predict.|