SavedModelBundle

מחלקה ציבורית SavedModelBundle

SavedModelBundle מייצג דגם שנטען מאחסון.

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

כיתות מקוננות

מַחלָקָה SavedModelBundle.Exporter אפשרויות לייצוא SavedModel.
מַחלָקָה SavedModelBundle.Loader אפשרויות לטעינת SavedModel.

קבועים

חוּט DEFAULT_TAG

שיטות ציבוריות

מפה<String, Tensor >
שיחה (מפה<String, Tensor > ארגומנטים)
מפעיל את פונקציית ברירת המחדל ישירות מדגם זה.
בָּטֵל
סגור ()
משחרר משאבים ( Graph Session ) המשויכים לחבילת המודלים השמורה.
סטטי SavedModelBundle.Exporter
יצואן (String exportDir)
ייצא דגם שמור.
ConcreteFunction
פונקציה (מחרוזת חתימת מפתח)
החזר ConcreteFunction המתאים לחתימת הפונקציה.
גרָף
גרף ()
מחזירה את הגרף שמתאר את החישוב שמבצע המודל.
סטטי SavedModelBundle
טען (String exportDir, String... תגיות)
טען דגם שמור מספריית ייצוא.
סטטי SavedModelBundle.Loader
מטעין (String exportDir)
טען דגם שמור.
MetaGraphDef
metaGraphDef ()
מחזירה את מאגר פרוטוקול MetaGraphDef המשויך למודל השמור.
מוֹשָׁב
מפגש ()
מחזירה את ה- Session שבאמצעותו יש לבצע חישוב באמצעות המודל.
רשימה< חתימה >
חתימות ()
החזר את החתימה של כל הפונקציות הזמינות בדגם השמור הזה.

שיטות בירושה

קבועים

מחרוזת סופית סטטית ציבורית DEFAULT_TAG

ערך קבוע: "הגשה"

שיטות ציבוריות

מפה ציבורית<String, Tensor > call (Map<String, Tensor > ארגומנטים)

מפעיל את פונקציית ברירת המחדל ישירות מדגם זה.

בחירת פונקציית ברירת המחדל נעשית על סמך הראשון מבין התנאים הבאים שהוא נכון:

  • הפונקציה היא החתימה היחידה הזמינה המצורפת לגרף הראשי של הדגם השמור הזה
  • הפונקציה ממופה לשם חתימת ברירת המחדל, שהוא "serving_default"

המתקשר אחראי לסגירת כל הטנזורים המוחזרים.

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

ריק ציבורי סגור ()

משחרר משאבים ( Graph Session ) המשויכים לחבילת המודלים השמורה.

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

public MetaGraphDef metaGraphDef ()

מחזירה את מאגר פרוטוקול MetaGraphDef המשויך למודל השמור.

מושב ציבורי ()

מחזירה את ה- Session שבאמצעותו יש לבצע חישוב באמצעות המודל.

מחזיר
  • הפגישה המותאמת

רשימה ציבורית< חתימה > חתימות ()

החזר את החתימה של כל הפונקציות הזמינות בדגם השמור הזה.