tfma.default_extractors

Returns the default extractors for use in ExtractAndEvaluate.

tfma.default_extractors(
    eval_shared_model: Optional[types.MaybeMultipleEvalSharedModels] = None,
    eval_config: tfma.EvalConfig = None,
    slice_spec: Optional[List[slicer.SingleSliceSpec]] = None,
    materialize: Optional[bool] = True,
    tensor_adapter_config: Optional[tensor_adapter.TensorAdapterConfig] = None,
    custom_predict_extractor: Optional[tfma.extractors.Extractor] = None
) -> List[tfma.extractors.Extractor]

eval_shared_model Shared model (single-model evaluation) or list of shared models (multi-model evaluation). Required unless the predictions are provided alongside of the features (i.e. model-agnostic evaluations).
eval_config Eval config.
slice_spec Deprecated (use EvalConfig).
materialize True to have extractors create materialized output.
tensor_adapter_config Tensor adapter config which specifies how to obtain tensors from the Arrow RecordBatch. If None, we feed the raw examples to the model.
custom_predict_extractor Optional custom predict extractor for non-TF models.

NotImplementedError If eval_config contains mixed serving and eval models.