|View source on GitHub|
Model loader is responsible for loading shared model types.
tfma.types.ModelLoader( tags: Optional[List[Text]] = None, shared_handle: Optional[shared.Shared] = None, construct_fn: Optional[Callable[..., Any]] = None )
||Model tags (e.g. 'serve' for serving or 'eval' for EvalSavedModel).|
||Optional handle to a shared.Shared object for sharing the in-memory model within / between stages. Used in combination with the construct_fn to load the shared model.|
||A callable which creates a construct function to load the model instance. Callable takes a beam.metrics distribution to track model load times.|