|View source on GitHub|
ModelCardToolkit provides utilities to generate a ModelCard.
model_card_toolkit.ModelCardToolkit( output_dir: Optional[Text] = None, mlmd_source: Optional[src.MlmdSource] = None, source: Optional[src.Source] = None )
ModelCardToolkit is a tool for ML practitioners to create Model Cards, documentation for model information such as owners, use cases, training and evaluation data, performance, etc. A Model Card document can be displayed in output formats including HTML, Markdown, etc.
The ModelCardToolkit includes an API designed for a human-in-the-loop process to elaborate the ModelCard. If model training is integrated with ML Metadata (e.g., TFX pipelines), the ModelCardToolkit can further populate ModelCard fields by extract metadata and lineage from the model's MLMD instance.
The ModelCardToolkit organizes the ModelCard assets (e.g., structured data, plots, and UI templates) in a user-specified directory, and updates them incrementally via its API.
import model_card_toolkit # Initialize the Model Card Toolkit with a path to store generate assets model_card_output_path = ... mct = model_card_toolkit.ModelCardToolkit(model_card_output_path) # Initialize the ModelCard, which can be freely populated model_card = mct.scaffold_assets() model_card.model_details.name = 'My Model' # Write the model card data to a proto file mct.update_model_card(model_card) # Return the model card document as an HTML page html = mct.export_format()
||The path where MCT assets (such as data files and model cards) are written to. If not provided, a temp directory is used.|
The ML Metadata Store to retrieve metadata and lineage
information about the model. If given, a set of model card properties
can be auto-populated from the
A collection of sources to extract data for a model card. This can
be used instead of
||If a model cannot be found at mlmd_source.model_uri.|
export_format( model_card: Optional[Union[
model_card_toolkit.ModelCard, model_card_pb2.ModelCard]] = None, template_path: Optional[Text] = None, output_file=_DEFAULT_MODEL_CARD_FILE_NAME ) -> Text
Generates a model card document based on the MCT assets.
The model card document is both returned by this function, as well as saved to output_file.
The ModelCard object, generated from
||The file path of the Jinja template. If not provided, the default template will be used.|
||The file name of the generated model card. If not provided, the default 'model_card.html' will be used. If the file already exists, then it will be overwritten.|
|The model card file content.|
save_mlmd() -> None
Saves the model card of the model artifact with
model_uri to MLMD.
Generates the Model Card Tookit assets.
Assets include the ModelCard proto file, Model Card document, and jinja
template. These are written to the
output_dir declared at
An assets directory is created if one does not already exist.
If the MCT is initialized with a
mlmd_store, it further auto-populates
the model cards properties as well as generating related plots such as model
performance and data distributions.
|A ModelCard representing the given model.|
||on failure to copy the template files.|
update_model_card( model_card: Union[
model_card_toolkit.ModelCard, model_card_pb2.ModelCard] ) -> None
Updates the Proto file in the MCT assets directory.
||The updated model card to write back.|
||when the given model_card is invalid w.r.t. the schema.|