Карты моделей в TFX

Компонент конвейера ModelCardGenerator TFX генерирует карточки моделей.

Подробный формат карты модели см. в API карты модели .

Более общую информацию о TFX см. в Руководстве пользователя TFX .

Настройка компонента ModelCardGenerator

ModelCardGenerator использует статистику набора данных , оценку модели и отправленную модель для автоматического заполнения частей карты модели.

Поля карточки модели также могут быть явно заполнены строкой JSON (ее можно сгенерировать с помощью модуля json , см. пример ниже). Если поле заполнено как TFX, так и JSON, значение JSON перезапишет значение TFX.

ModelCardGenerator записывает документы карточек моделей в каталог model_card/ выходных данных артефакта. Он использует шаблон карты модели HTML по умолчанию, который используется для создания model_card.html . Также можно использовать собственные шаблоны ; каждый ввод шаблона должен сопровождаться выводом имени файла в аргументе template_io .

Пример

from model_card_toolkit import ModelCardGenerator
import json

...
model_card_fields = {
  'model_details': {
    'name': 'my_model',
    'owners': 'Google',
    'version': 'v0.1'
  },
  'considerations': {
    'limitations': 'This is a demo model.'
  }
}
mc_gen = ModelCardGenerator(
    statistics=statistics_gen.outputs['statistics'],
    evaluation=evaluator.outputs['evaluation'],
    pushed_model=pusher.outputs['pushed_model'],
    json=json.dumps(model_card_fields),
    template_io=[
        ('html/default_template.html.jinja', 'model_card.html'),
        ('md/default_template.md.jinja', 'model_card.md')
    ]
)

Более подробную информацию можно найти в справочнике по API ModelCardGenerator .

Полный рабочий пример смотрите в нашей комплексной демонстрации .