קונספטים של ערכת כלים לכרטיסי דגם

כרטיס דגם

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

כמה מקרי שימוש של כרטיסי דגם כוללים:

  • הקלת חילופי מידע בין בוני מודלים ומפתחי מוצר.
  • ליידע את המשתמשים במודלים של ML לקבל החלטות מושכלות יותר לגבי אופן השימוש בהם (או כיצד לא להשתמש בהם).
  • מתן מידע מודל הנדרש לפיקוח ציבורי יעיל ואחריות.

סכֵימָה

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

כיום, סכימת כרטיס הדגם נאכפת בקפדנות. ב-Model Card Toolkit 2.0, מגבלת סכימה זו תוסר.

גרָפִיקָה

Model Card Toolkit מייצר אוטומטית גרפיקה עבור מערכי נתונים של TFX ותוצאות הערכה. ניתן גם ליצור גרפיקה ידנית באמצעות כלי כמו Matplotlib, ולכתוב ל-ModelCard - ראה ממשק API של כרטיס דגם לפרטים.

בסכימת כרטיס הדגם, גרפיקה מאוחסנת בשדה Graphic.image ומקודדת כמחרוזות מקודדות base64 . ערכת הכלים לכרטיסי מודל יכולה לעזור ביצירת תמונות base64 .

ערכת כלים לכרטיסי דגם

ערכת הכלים של כרטיס דגם מאפשרת לך ליצור מסמכי כרטיס דגם , כמו גם אובייקטי פרוטו ו-JSON , עם ממשק Python יעיל.

API של כרטיס דגם

ערכת הכלים לכרטיסי מודל כוללת API של כרטיס דגם המורכב ממחלקת Python . עדכונים שנעשו לאובייקט Python של כרטיס Model נכתבים לאובייקט פרוטו של כרטיס Model.

גרָפִיקָה

ניתן להשתמש בפונקציה model_card_toolkit.utils.graphics.figure_to_base64str() כדי להמיר גרפיקה, כגון דמויות Matplotlib, למחרוזות base64.

שמירה וטעינה של כרטיסי דגם

אם סיימת להוסיף הערות לכרטיס הדגם שלך וברצונך להעביר אותו בסידרה בפורמט JSON או protobuf, השתמש בשיטה ModelCard.save() .


import model_card_toolkit as mct

model_card = mct.ModelCard()
model_card.model_details.name = 'Fine-tuned MobileNetV2 Model for Cats vs. Dogs'
model_card.save('model_cards/cats_vs_dogs.json')

אם תרצה לשחזר ולעדכן כרטיס דגם שמור, השתמש בפונקציה model_card_toolkit.model_card.load_model_card() .


import model_card_toolkit as mct

model_card = mct.load_model_card('model_cards/cats_vs_dogs.json')
model_card.model_details.licenses.append(mct.License(identifier='Apache-2.0'))

מסמכי כרטיס דגם

כברירת מחדל, מסמך כרטיס הדגם שנוצר הוא קובץ HTML המבוסס על default_template.html.jinja . עם זאת, אתה יכול גם לספק תבנית Jinja מותאמת אישית משלך. קבצי תבניות אלה יכולים להיות כל פורמט מבוסס טקסט (HTML, Markdown, LaTeX וכו'). תבנית Markdown מסופקת כדוגמה.

שילוב TFX ו-MLMD

ערכת הכלים לכרטיסי דגם משתלבת עם הכלים TensorFlow Extended ו- ML Metadata . ניתן להשתמש ב-Metadata Store במהלך אתחול ערכת הכלים של כרטיס מודל כדי לאכלס מראש שדות רבים של כרטיסי מודל וליצור מגרשי הדרכה והערכה.

חפצים המשמשים את MCT:

ביצועים המשמשים את MCT:

  • מאמן : משמש לאכלוס שם הדגם והגרסה

רכיב ModelCardGenerator עבר לספריית התוספות של TFX ואינו ארוז עוד ב-Model Card Toolkit מגרסה 2.0.0. לפני שתוכל להשתמש ברכיב, תצטרך להתקין את חבילת tfx-addons :

pip install tfx-addons[model_card_generator]

עיין במדריך ModelCardGenerator והפעל את מחברת המקרה כדי ללמוד עוד על הרכיב.