Le composant de pipeline ModelCardGenerator TFX génère des cartes modèles.
Pour le format détaillé de la carte modèle, consultez l' API Model Card .
Pour des informations plus générales sur TFX, veuillez consulter le Guide de l'utilisateur TFX .
Configuration du composant ModelCardGenerator
Le ModelCardGenerator prend des statistiques d'ensemble de données , une évaluation de modèle et un modèle poussé pour remplir automatiquement certaines parties d'une carte de modèle.
Les champs de la carte modèle peuvent également être explicitement renseignés avec une chaîne JSON (celle-ci peut être générée à l'aide du module json , voir l'exemple ci-dessous). Si un champ est renseigné à la fois par TFX et JSON, la valeur JSON écrasera la valeur TFX.
ModelCardGenerator écrit les documents de carte modèle dans le répertoire model_card/ de sa sortie d'artefact. Il utilise un modèle de carte modèle HTML par défaut, qui est utilisé pour générer model_card.html . Des modèles personnalisés peuvent également être utilisés ; chaque entrée de modèle doit être accompagnée d'un nom de fichier en sortie dans l'argument template_io .
Exemple
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')
]
)
Plus de détails sont disponibles dans la référence de l'API ModelCardGenerator .
Consultez notre démo de bout en bout pour un exemple de travail complet.