คำถามเกี่ยวกับ TFX? เข้าร่วมกับเราที่ Google I / O!

Pusher TFX Pipeline Component

คอมโพเนนต์ Pusher ใช้เพื่อพุชโมเดลที่ตรวจสอบแล้วไปยัง เป้าหมายการปรับใช้ ระหว่างการฝึกโมเดลหรือการฝึกซ้ำ ก่อนการปรับใช้ Pusher อาศัยพรอย่างน้อยหนึ่งอย่างจากองค์ประกอบการตรวจสอบความถูกต้องอื่น ๆ เพื่อตัดสินใจว่าจะผลักดันโมเดลหรือไม่

  • ผู้ประเมินจะ อวยพรโมเดลหากโมเดลที่ได้รับการฝึกฝนใหม่ "ดีพอ" ที่จะถูกผลักดันไปสู่การผลิต
  • (ทางเลือก แต่แนะนำ) InfraValidator เป็น พรแก่โมเดลหากโมเดลนั้นสามารถใช้งานได้โดยกลไกในสภาพแวดล้อมการใช้งานจริง

คอมโพเนนต์ Pusher ใช้โมเดลที่ผ่านการฝึกอบรมในรูปแบบ SavedModel และสร้าง SavedModel เดียวกันพร้อมกับข้อมูลเมตาการกำหนดเวอร์ชัน

การใช้ Pusher Component

โดยทั่วไปแล้วส่วนประกอบ Pusher ไปป์ไลน์นั้นง่ายมากในการปรับใช้และต้องการการปรับแต่งเพียงเล็กน้อยเนื่องจากงานทั้งหมดทำโดยส่วนประกอบ Pusher TFX รหัสทั่วไปมีลักษณะดังนี้:

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