tfma.types.ModelLoader

Model loader is responsible for loading shared model types.

tfma.types.ModelLoader(
    construct_fn: Callable[[], Any], tags: Optional[List[str]] = None
)

construct_fn A callable which creates the model instance. The callable should take no args as input (typically a closure is used to capture necessary parameters).
tags Optional model tags (e.g. 'serve' for serving or 'eval' for EvalSavedModel).

Methods

load

View source

load(
    model_load_time_callback: Optional[Callable[[int], None]] = None
) -> Any

Returns loaded model.

Args
model_load_time_callback Optional callback to track load time.