Cartões de modelo no TFX

O componente de pipeline ModelCardGenerator TFX gera cartões de modelo.

Para obter o formato detalhado do cartão de modelo, consulte API de cartão de modelo .

Para obter informações mais gerais sobre o TFX, consulte o Guia do usuário do TFX .

Configurando o componente ModelCardGenerator

O ModelCardGenerator usa estatísticas do conjunto de dados , avaliação de modelo e um modelo enviado para preencher automaticamente partes de um cartão de modelo.

Os campos do cartão modelo também podem ser preenchidos explicitamente com uma string JSON (isso pode ser gerado usando o módulo json , veja o exemplo abaixo). Se um campo for preenchido por TFX e JSON, o valor JSON substituirá o valor TFX.

O ModelCardGenerator grava documentos de cartão modelo no diretório model_card/ de sua saída de artefato. Ele usa um modelo de cartão HTML padrão, que é usado para gerar model_card.html . Modelos personalizados também podem ser usados; cada entrada de modelo deve ser acompanhada por uma saída de nome de arquivo no argumento template_io .

Exemplo

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')
    ]
)

Mais detalhes estão disponíveis na referência da API ModelCardGenerator .

Veja nossa demonstração completa para um exemplo completo de trabalho.