¿Preguntas sobre TFX? ¡Únase a nosotros en Google I / O!

El componente de tubería Pusher TFX

El componente Pusher se utiliza para enviar un modelo validado a un destino de implementación durante el entrenamiento o reentrenamiento del modelo. Antes de la implementación, Pusher se basa en una o más ventajas de otros componentes de validación para decidir si impulsar el modelo o no.

  • El evaluador bendice el modelo si el nuevo modelo entrenado es "lo suficientemente bueno" para ser impulsado a producción.
  • (Opcional pero recomendado) InfraValidator bendice el modelo si el modelo se puede servir mecánicamente en un entorno de producción.

Un componente Pusher consume un modelo entrenado en formato SavedModel y produce el mismo SavedModel, junto con metadatos de control de versiones.

Uso del componente Pusher

Un componente de canalización de Pusher suele ser muy fácil de implementar y requiere poca personalización, ya que todo el trabajo lo realiza el componente Pusher TFX. El código típico tiene este aspecto:

from tfx import components

...

pusher = components.Pusher(
  model=trainer.outputs['model'],
  model_blessing=evaluator.outputs['blessing'],
  infra_blessing=infra_validator.outputs['blessing'],
  push_destination=pusher_pb2.PushDestination(
    filesystem=pusher_pb2.PushDestination.Filesystem(
        base_directory=serving_model_dir)
  )
)