TensorFlow Hub là một kho lưu trữ và thư viện mở để máy học có thể tái sử dụng. Kho lưu trữ tfhub.dev cung cấp nhiều mô hình được đào tạo trước: nhúng văn bản, mô hình phân loại hình ảnh, mô hình TF.js / TFLite và hơn thế nữa. Kho lưu trữ được mở cho những người đóng góp trong cộng đồng .
Thư viện tensorflow_hub
cho phép bạn tải xuống và sử dụng lại chúng trong chương trình TensorFlow của mình với số lượng mã tối thiểu.
import tensorflow_hub as hub
model = hub.KerasLayer("https://tfhub.dev/google/nnlm-en-dim128/2")
embeddings = model(["The rain in Spain.", "falls",
"mainly", "In the plain!"])
print(embeddings.shape) #(4,128)