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(
    tags: Optional[List[Text]] = None,
    shared_handle: Optional[shared.Shared] = None,
    construct_fn: Optional[Callable[..., Any]] = None
)

Attributes
tags	Model tags (e.g. 'serve' for serving or 'eval' for EvalSavedModel).
shared_handle	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.
construct_fn	A callable which creates a construct function to load the model instance. Callable takes a beam.metrics distribution to track model load times.