tfma.types.ModelLoader

View source on GitHub

Model loader is responsible for loading shared model types.

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.