TensorFlow Hub - это библиотека для многоразовых модулей машинного обучения.
!pip install "tensorflow_hub>=0.6.0"
!pip install "tensorflow>=2.0.0"
import tensorflow as tf
import tensorflow_hub as hub
module_url = "https://tfhub.dev/google/nnlm-en-dim128/2"
embed = hub.KerasLayer(module_url)
embeddings = embed(["A long sentence.", "single-word",
"http://example.com"])
print(embeddings.shape) #(3,128)
TensorFlow Hub - это библиотека для публикации, обнаружения и использования повторно используемых частей моделей машинного обучения. Модуль - это автономный фрагмент графа TensorFlow вместе с его весами и активами, который можно повторно использовать для решения различных задач в процессе, известном как трансферное обучение. Трансферное обучение может:
- Обучите модель с меньшим набором данных,
- Улучшить обобщение и
- Ускорьте обучение.
