TensorFlow 2.0 Beta is available Learn more

tfma.default_extractors

tfma.default_extractors(
    eval_shared_model,
    slice_spec=None,
    desired_batch_size=None,
    materialize=True
)

Defined in api/model_eval_lib.py.

Returns the default extractors for use in ExtractAndEvaluate.

Args:

  • eval_shared_model: Shared model parameters for EvalSavedModel.
  • slice_spec: Optional list of SingleSliceSpec specifying the slices to slice the data into. If None, defaults to the overall slice.
  • desired_batch_size: Optional batch size for batching in Aggregate.
  • materialize: True to have extractors create materialized output.