tfma.types.ModelLoader

View source on GitHub

Model loader is responsible for loading shared model types.

View aliases

Main aliases

tfma.ModelLoader

tfma.types.ModelLoader(
    construct_fn: Callable[[], Any],
    tags: Optional[List[Text]] = 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

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.

Class Variables

• construct_fn
• tags