GitHub

tfma.types.ModelLoader

Model loader is responsible for loading shared model types.

Main aliases

tfma.ModelLoader

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

Attributes
`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(
    model_load_time_callback: Optional[Callable[[int], None]] = None
) -> Any

Returns loaded model.

Args
`model_load_time_callback`	Optional callback to track load time.

Except as otherwise noted, the content of this page is licensed under the Creative Commons Attribution 4.0 License, and code samples are licensed under the Apache 2.0 License. For details, see the Google Developers Site Policies. Java is a registered trademark of Oracle and/or its affiliates.

Last updated 2022-05-20 UTC.