סקירה כללית של ספריית TensorFlow Hub

tensorflow_hub הספרייה מאפשרת לך להוריד ולעשות שימוש חוזר במודלים הנלמדים בתוכנית TensorFlow שלך עם כמות מינימלית של קוד. הדרך העיקרית לטעון מודל מודרך משתמש hub.KerasLayer API.

import tensorflow_hub as hub

embed = hub.KerasLayer("https://hub.tensorflow.google.cn/google/nnlm-en-dim128/2")
embeddings = embed(["A long sentence.", "single-word", "http://example.com"])
print(embeddings.shape, embeddings.dtype)

הגדרת מיקום המטמון להורדות.

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

יציבות API

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

הֲגִינוּת

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

בִּטָחוֹן

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

הצעדים הבאים