Model loader is responsible for loading shared model types.
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
|
|