רכיב הצינור 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.
עיין בהדגמה שלנו מקצה לקצה לקבלת דוגמה מלאה לעבודה.