Tarjetas modelo en TFX

El componente de canalización ModelCardGenerator TFX genera tarjetas modelo.

Para conocer 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 enviado 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 con TFX y JSON, el valor JSON sobrescribirá el valor TFX.

ModelCardGenerator escribe documentos de tarjeta modelo en el directorio model_card/ de su salida de artefacto. Utiliza una plantilla de tarjeta 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')
    ]
)

Más detalles están disponibles en la referencia de la API ModelCardGenerator .

Vea nuestra demostración de un extremo a otro para ver un ejemplo funcional completo.