TFX의 모델 카드

ModelCardGenerator TFX 파이프라인 구성요소는 모델 카드를 생성합니다.

자세한 모델 카드 형식은 모델 카드 API를 참조하세요.

TFX에 대한 일반적인 정보는 TFX 사용자 가이드를 참조하세요.

ModelCardGenerator 구성 요소 구성

ModelCardGenerator는 데이터세트 통계 , 모델 평가 , 푸시된 모델을 사용하여 모델 카드의 일부를 자동으로 채웁니다.

모델 카드 필드는 JSON 문자열로 명시적으로 채워질 수도 있습니다(이는 json 모듈을 사용하여 생성할 수 있습니다. 아래 예 참조). 필드가 TFX와 JSON 모두로 채워지면 JSON 값이 TFX 값을 덮어씁니다.

ModelCardGenerator는 모델 카드 문서를 해당 아티팩트 출력의 model_card/ 디렉터리에 기록합니다. model_card.html 생성하는 데 사용되는 기본 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')
    ]
)

자세한 내용은 ModelCardGenerator API 참조에서 확인할 수 있습니다.

전체 작업 예제를 보려면 엔드투엔드 데모를 참조하세요.