כרטיסי דגם ב-TFX

רכיב הצינור ModelCardGenerator TFX מייצר כרטיסי דגם.

לפורמט כרטיס הדגם המפורט, עיין ב- API של כרטיס הדגם .

למידע כללי נוסף על TFX, עיין במדריך למשתמש של TFX .

הגדרת רכיב ModelCardGenerator

ModelCardGenerator לוקח נתונים סטטיסטיים , הערכת מודל ומודל נדחף כדי לאכלס באופן אוטומטי חלקים מכרטיס דגם.

שדות כרטיס דגם יכולים גם להיות מאוכלסים במפורש במחרוזת JSON (ניתן ליצור זאת באמצעות מודול json , ראה דוגמה למטה). אם שדה מאוכלס ב-TFX וגם ב-JSON, ערך ה-JSON יחליף את ערך ה-TFX.

ModelCardGenerator כותב מסמכי כרטיס דגם לספריית model_card/ של פלט החפצים שלו. הוא משתמש בתבנית ברירת המחדל של כרטיס מודל HTML, המשמשת ליצירת model_card.html . ניתן להשתמש גם בתבניות מותאמות אישית; כל קלט של תבנית חייב להיות מלווה בפלט שם קובץ ב-arg template_io .

דוגמא

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.

עיין בהדגמה שלנו מקצה לקצה לקבלת דוגמה מלאה לעבודה.