Unisciti alla comunità SIG TFX-Addons e contribuisci a rendere TFX ancora migliore!

Il componente Pusher TFX Pipeline

Il componente Pusher viene utilizzato per inviare un modello convalidato a una destinazione di distribuzione durante l'addestramento o il ri-addestramento del modello. Prima della distribuzione, Pusher fa affidamento su una o più benedizioni di altri componenti di convalida per decidere se eseguire il push del modello o meno.

  • Il valutatore benedice il modello se il nuovo modello addestrato è "abbastanza buono" per essere spinto alla produzione.
  • (Facoltativo ma consigliato) InfraValidator benedice il modello se il modello è utilizzabile meccanicamente in un ambiente di produzione.

Un componente Pusher utilizza un modello addestrato in formato SavedModel e produce lo stesso SavedModel, insieme ai metadati di controllo delle versioni.

Utilizzo del componente Pusher

Un componente pipeline Pusher è in genere molto facile da distribuire e richiede poca personalizzazione, poiché tutto il lavoro viene svolto dal componente Pusher TFX. Il codice tipico ha questo aspetto:

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

Maggiori dettagli sono disponibili nel riferimento API Pusher .