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.