Module core.components.evaluator.evaluator_module

Functions

BatchedPredictExtractor(eval_config: tensorflow_model_analysis.proto.config_pb2.EvalConfig, eval_shared_model: Union[tensorflow_model_analysis.types.EvalSharedModel, List[tensorflow_model_analysis.types.EvalSharedModel], Dict[str, tensorflow_model_analysis.types.EvalSharedModel]], tensor_adapter_config: Union[tfx_bsl.tfxio.tensor_adapter.TensorAdapterConfig, NoneType] = None) ‑> tensorflow_model_analysis.extractors.extractor.Extractor :

custom_extractors(eval_config, eval_shared_model, tensor_adapter_config) ‑> List[tensorflow_model_analysis.extractors.extractor.Extractor] :