El componente de canalización ModelCardGenerator TFX genera tarjetas modelo.
Para ver el formato detallado de la tarjeta modelo, consulte la API de tarjeta modelo .
Para obtener más información general sobre TFX, consulte la Guía del usuario de TFX .
Configuración del componente ModelCardGenerator
ModelCardGenerator toma estadísticas de conjuntos de datos, evaluación de modelos y un modelo insertado para completar automáticamente partes de una tarjeta de modelo.
Los campos de la tarjeta modelo también se pueden completar explícitamente con una cadena JSON (esto se puede generar usando el módulo json , consulte el ejemplo a continuación). Si un campo se completa tanto con TFX como con JSON, el valor de JSON sobrescribirá el valor de TFX.
ModelCardGenerator escribe documentos de tarjeta modelo en el directorio model_card/ de su salida de artefacto. Utiliza una plantilla de tarjeta de modelo HTML predeterminada, que se utiliza para generar model_card.html . También se pueden utilizar plantillas personalizadas; cada entrada de plantilla debe ir acompañada de una salida de nombre de archivo en template_io arg.
Ejemplo
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')
]
)
Hay más detalles disponibles en la referencia de la API ModelCardGenerator .
Vea nuestra demostración integral para ver un ejemplo de trabajo completo.