מחלקה ציבורית SavedModelBundle
SavedModelBundle מייצג דגם שנטען מאחסון.
המודל מורכב מתיאור של החישוב ( Graph ), Session עם טנסורים (למשל, פרמטרים או משתנים בגרף) המאוחל לערכים שנשמרו באחסון, ותיאור של המודל כמאגר פרוטוקול MetaGraphDef .
כיתות מקוננות
| מַחלָקָה | SavedModelBundle.Exporter | אפשרויות לייצוא SavedModel. | |
| מַחלָקָה | SavedModelBundle.Loader | אפשרויות לטעינת SavedModel. | |
קבועים
| חוּט | DEFAULT_TAG |
שיטות ציבוריות
| מפה<String, Tensor > | |
| בָּטֵל | |
| סטטי SavedModelBundle.Exporter | יצואן (String exportDir) ייצא דגם שמור. |
| ConcreteFunction | פונקציה (מחרוזת חתימת מפתח) החזר ConcreteFunction המתאים לחתימת הפונקציה. |
| גרָף | גרף () מחזירה את הגרף שמתאר את החישוב שמבצע המודל. |
| סטטי SavedModelBundle | טען (String exportDir, String... תגיות) טען דגם שמור מספריית ייצוא. |
| סטטי SavedModelBundle.Loader | מטעין (String exportDir) טען דגם שמור. |
| MetaGraphDef | metaGraphDef () מחזירה את מאגר פרוטוקול MetaGraphDef המשויך למודל השמור. |
| מוֹשָׁב | |
| רשימה< חתימה > | חתימות () החזר את החתימה של כל הפונקציות הזמינות בדגם השמור הזה. |
שיטות בירושה
קבועים
מחרוזת סופית סטטית ציבורית DEFAULT_TAG
ערך קבוע: "הגשה"
שיטות ציבוריות
מפה ציבורית<String, Tensor > call (Map<String, Tensor > ארגומנטים)
מפעיל את פונקציית ברירת המחדל ישירות מדגם זה.
בחירת פונקציית ברירת המחדל נעשית על סמך הראשון מבין התנאים הבאים שהוא נכון:
- הפונקציה היא החתימה היחידה הזמינה המצורפת לגרף הראשי של הדגם השמור הזה
- הפונקציה ממופה לשם חתימת ברירת המחדל, שהוא "serving_default"
המתקשר אחראי לסגירת כל הטנזורים המוחזרים.
פרמטרים
| טיעונים | רשימה של טנסור קלט, ממופה לפי שם החתימה שלהם |
|---|
מחזיר
- רשימה של טנסור פלט, ממופה לפי שם החתימה
זורק
| חריג טיעון לא חוקי | אם לא ניתן לבחור פונקציה כברירת מחדל |
|---|
יצואנית SavedModelBundle.Exporter סטטית ציבורית (String exportDir)
ייצא דגם שמור.
מחזירה אובייקט Exporter להגדרת אפשרויות תצורה לפני שמירת המודל בפועל.
פרמטרים
| exportDir | נתיב הספרייה המכיל מודל שמור. |
|---|
פונקציית ConcreteFunction ציבורית (מחרוזת חתימת מפתח)
החזר ConcreteFunction המתאים לחתימת הפונקציה.
ConcreteFunction myFunction = savedModelBundle.function("mySignatureKey");
Map<String, Tensor> outputTensorMap = myFunction.call(session, inputTensorMap);
פרמטרים
| מפתח חתימה | שם ה- SignatureDef במודל השמור. |
|---|
מחזיר
- אובייקט שניתן להשתמש בו כדי לבצע קריאות לפונקציה
זורק
| חריג טיעון לא חוקי | אם signatureKey לא נמצא בדגם השמור הזה. |
|---|
טעינת חבילת SavedModelBundle סטטית ציבורית (String exportDir, String... תגיות)
טען דגם שמור מספריית ייצוא. יש ליצור את המודל הנטען באמצעות ה- API של מודל שמור .
שיטה זו היא קיצור של:
SavedModelBundle.loader().withTags(tags).load();
פרמטרים
| exportDir | נתיב הספרייה המכיל מודל שמור. |
|---|---|
| תגים | התגים המזהים את metagraphdef הספציפי לטעינה. |
מחזיר
- חבילה המכילה את הגרף וההפעלה המשויכת.
מטעין SavedModelBundle.Loader סטטי ציבורי (String exportDir)
טען דגם שמור.
מחזירה אובייקט Loader שיכול להגדיר אפשרויות תצורה לפני טעינת המודל בפועל,
פרמטרים
| exportDir | נתיב הספרייה המכיל מודל שמור. |
|---|