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.