การ์ดโมเดลใน TFX

ส่วนประกอบไปป์ไลน์ ModelCardGenerator TFX จะสร้างการ์ดโมเดล

สำหรับรูปแบบการ์ดโมเดลโดยละเอียด โปรดดูที่ Model Card API

สำหรับข้อมูลทั่วไปเพิ่มเติมเกี่ยวกับ TFX โปรดดู คู่มือผู้ใช้ TFX

การกำหนดค่าส่วนประกอบ ModelCardGenerator

ModelCardGenerator ใช้ สถิติชุดข้อมูล การประเมินโมเดล และ โมเดลที่พุช เพื่อเติมข้อมูลบางส่วนของการ์ดโมเดลโดยอัตโนมัติ

ช่องการ์ดโมเดล สามารถเติมข้อมูลอย่างชัดเจนด้วยสตริง JSON ได้ (ซึ่งสามารถสร้างได้โดยใช้โมดูล json ดูตัวอย่างด้านล่าง) หากช่องป้อนข้อมูลด้วย TFX และ JSON ค่า JSON จะเขียนทับค่า TFX

ModelCardGenerator เขียนเอกสารการ์ดโมเดลลงในไดเร็กทอรี model_card/ ของเอาต์พุตสิ่งประดิษฐ์ โดยจะใช้เทมเพลตการ์ดโมเดล HTML เริ่มต้น ซึ่งใช้ในการสร้าง model_card.html สามารถใช้ เทมเพลต แบบกำหนดเองได้ อินพุตเทมเพลตแต่ละรายการจะต้องมาพร้อมกับเอาต์พุตชื่อไฟล์ใน template_io arg

ตัวอย่าง

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

รายละเอียดเพิ่มเติมมีอยู่ในข้อมูลอ้างอิง ModelCardGenerator API

ดู การสาธิตแบบ end-to-end ของเราสำหรับตัวอย่างการทำงานแบบเต็ม