O componente de pipeline ModelCardGenerator TFX gera cartões de modelo.
Para o formato de cartão de modelo detalhado, consulte a API do cartão de modelo .
Para obter mais informações gerais sobre o TFX, consulte o Guia do usuário do TFX .
Configurando o componente ModelCardGenerator
O ModelCardGenerator usa estatísticas de 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 de 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 de modelo no diretório model_card/ de sua saída de artefato. Ele usa um modelo de cartão de modelo 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 template_io arg.
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 de ponta a ponta para um exemplo completo de trabalho.