ไลบรารี tensorflow_hub ให้คุณดาวน์โหลดและนำโมเดลที่ผ่านการฝึกอบรมกลับมาใช้ซ้ำในโปรแกรม TensorFlow ของคุณด้วยโค้ดขั้นต่ำ วิธีหลักในการโหลดโมเดลที่ผ่านการฝึกอบรมแล้วคือการใช้ hub.KerasLayer API
import tensorflow_hub as hub
embed = hub.KerasLayer("https://tfhub.dev/google/nnlm-en-dim128/2")
embeddings = embed(["A long sentence.", "single-word", "http://example.com"])
print(embeddings.shape, embeddings.dtype)
หมายเหตุ: เอกสารนี้ใช้ตัวจัดการ URL ของ TFhub.dev ในตัวอย่าง ดูข้อมูลเพิ่มเติมเกี่ยวกับประเภทด้ามจับที่ถูกต้องอื่นๆ ที่นี่
การตั้งค่าตำแหน่งแคชสำหรับการดาวน์โหลด
ตามค่าเริ่มต้น tensorflow_hub จะใช้ไดเร็กทอรีชั่วคราวทั้งระบบเพื่อแคชโมเดลที่ดาวน์โหลดและไม่มีการบีบอัด ดู การแคช สำหรับตัวเลือกในการใช้ตำแหน่งอื่นที่อาจถาวรมากกว่า
ความเสถียรของเอพีไอ
แม้ว่าเราหวังว่าจะป้องกันไม่ให้เกิดการเปลี่ยนแปลง แต่โปรเจ็กต์นี้ยังอยู่ระหว่างการพัฒนาและยังไม่รับประกันว่าจะมี API หรือรูปแบบโมเดลที่เสถียร
ความเป็นธรรม
เช่นเดียวกับ Machine Learning ทั้งหมด ความเป็นธรรม คือการพิจารณา ที่สำคัญ โมเดลที่ได้รับการฝึกล่วงหน้าหลายตัวได้รับการฝึกบนชุดข้อมูลขนาดใหญ่ เมื่อนำโมเดลใดๆ มาใช้ซ้ำ สิ่งสำคัญคือต้องคำนึงถึงข้อมูลที่โมเดลนั้นได้รับการฝึก (และมีความเอนเอียงใดๆ ที่มีอยู่หรือไม่) และสิ่งเหล่านี้อาจส่งผลต่อการใช้งานของคุณอย่างไร
ความปลอดภัย
เนื่องจากมีกราฟ TensorFlow ที่กำหนดเอง แบบจำลองจึงถือเป็นโปรแกรมได้ การใช้ TensorFlow อย่างปลอดภัย จะอธิบายผลกระทบด้านความปลอดภัยของการอ้างอิงโมเดลจากแหล่งที่ไม่น่าเชื่อถือ