Model Card Toolkit
Model Card Toolkit
The Model Card Toolkit (MCT) streamlines and automates generation of Model Cards , machine learning documents that provide context and transparency into a model's development and performance. Integrating the MCT into your ML pipeline enables the sharing model metadata and metrics with researchers, developers, reporters, and more.
Some use cases of model cards include:
- Facilitating the exchange of information between model builders and product developers.
- Informing users of ML models to make better-informed decisions about how to use them (or how not to use them).
- Providing model information required for effective public oversight and accountability.
The Model Card Toolkit is hosted on PyPI, and can be installed with
pip install model-card-toolkit (or
pip install model-card-toolkit --use-deprecated=legacy-resolver for pip20.3). See the installation guide for more details.
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 model_card_toolkit.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 JSON file mct.update_model_card_json(model_card) # Return the model card document as an HTML page html = mct.export_format()
Automatic Model Card Generation
If your machine learning pipeline uses the TensorFlow Extended (TFX) platform or ML Metadata, you can automate model card generation. See this demo notebook for a demonstration of how to integrate the MCT into your pipeline.
Model cards are stored in JSON as an intermediate format. You can see the model card JSON schema in the
schema directory. Note that this is not a finalized path and may be hosted elsewhere in the future.
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
|Filename, size||File type||Python version||Upload date||Hashes|
|Filename, size model_card_toolkit-0.1.3-py3-none-any.whl (44.0 kB)||File type Wheel||Python version py3||Upload date||Hashes View|
|Filename, size model-card-toolkit-0.1.3.tar.gz (30.2 kB)||File type Source||Python version None||Upload date||Hashes View|
Hashes for model_card_toolkit-0.1.3-py3-none-any.whl