Treten Sie der SIG TFX-Addons-Community bei und helfen Sie, TFX noch besser zu machen!

Die Pusher TFX-Pipeline-Komponente

Die Pusher-Komponente wird verwendet, um ein validiertes Modell während des Modelltrainings oder des Re-Trainings an ein Bereitstellungsziel zu senden . Vor der Bereitstellung verlässt sich Pusher auf einen oder mehrere Segnungen anderer Validierungskomponenten, um zu entscheiden, ob das Modell gepusht werden soll oder nicht.

  • Der Evaluator segnet das Modell, wenn das neu trainierte Modell "gut genug" ist, um in die Produktion gebracht zu werden.
  • (Optional, aber empfohlen) InfraValidator segnet das Modell, wenn das Modell in einer Produktionsumgebung mechanisch gewartet werden kann.

Eine Pusher-Komponente verwendet ein trainiertes Modell im SavedModel- Format und erstellt dasselbe SavedModel zusammen mit Versionsmetadaten.

Verwenden der Drückerkomponente

Eine Pusher-Pipeline-Komponente ist in der Regel sehr einfach bereitzustellen und erfordert nur wenige Anpassungen, da die gesamte Arbeit von der Pusher TFX-Komponente ausgeführt wird. Typischer Code sieht folgendermaßen aus:

pusher = 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)
  )
)

Weitere Details finden Sie in der Pusher-API-Referenz .