Il componente pipeline ModelCardGenerator TFX genera schede modello.
Per il formato dettagliato della scheda modello, consulta l' API della scheda modello .
Per informazioni più generali su TFX, consultare la Guida per l'utente di TFX .
Configurazione del componente ModelCardGenerator
ModelCardGenerator utilizza le statistiche del set di dati , la valutazione del modello e un modello inviato per popolare automaticamente le parti di una scheda modello.
I campi della scheda modello possono anche essere popolati esplicitamente con una stringa JSON (questa può essere generata utilizzando il modulo json , vedere l'esempio di seguito). Se un campo è popolato sia da TFX che da JSON, il valore JSON sovrascriverà il valore TFX.
Il ModelCardGenerator scrive i documenti della scheda modello nella directory model_card/ del suo output artefatto. Utilizza un modello di scheda modello HTML predefinito, che viene utilizzato per generare model_card.html . È possibile utilizzare anche modelli personalizzati; ogni input del modello deve essere accompagnato dall'output di un nome file nell'argomento template_io .
Esempio
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')
]
)
Maggiori dettagli sono disponibili nel riferimento API ModelCardGenerator .
Guarda la nostra demo end-to-end per un esempio completo e funzionante.