模型卡工具包

模型卡工具包 (MCT) 库简化并自动化了 模型卡 的生成，模型卡是机器学习文档，提供模型开发和性能的背景和透明度。将模型卡工具包集成到您的 ML 管道中，可以让您与研究人员、开发人员、记者等共享模型的元数据和指标。

MCT 使用 JSON 架构 存储模型卡字段。MCT 可以通过 ML 元数据 (MLMD) 为 TFX 用户自动填充这些字段。模型卡字段也可以通过 Python API 手动填充。模型卡的一些用例包括

  • 促进模型构建者和产品开发人员之间信息交流。
  • 告知 ML 模型的用户，让他们能够更明智地决定如何使用（或不使用）这些模型。
  • 提供有效公共监督和问责制所需的模型信息。
import model_card_toolkit as mct

# Initialize the Model Card Toolkit with a path to store generate assets
model_card_output_path = ...
toolkit = mct.ModelCardToolkit(model_card_output_path)

# Initialize the ModelCard, which can be freely populated
model_card = toolkit.scaffold_assets()
model_card.model_details.name = 'My Model'

# Write the model card data to a JSON file
toolkit.update_model_card_json(model_card)

# Return the model card document as an HTML page
html = toolkit.export_format()

资源

尝试笔记本教程
尝试笔记本教程
在 Google AI 博客上阅读