tfma.types.ModelLoader

View source on GitHub

Model loader is responsible for loading shared model types.

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

Returns loaded model.

Args
model_load_time_callback Optional callback to track load time.

Class Variables

  • construct_fn
  • tags